labelmeWithAI/labelme/cli/draw_label_png.py

import argparse
import logging

import matplotlib.pyplot as plt
import numpy as np
import PIL.Image

from labelme import utils


def main():
    logger = logging.Logger('labelme')
    logger.setLevel(logging.INFO)

    parser = argparse.ArgumentParser(
        formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument('label_png', help='label PNG file')
    args = parser.parse_args()

    lbl = np.asarray(PIL.Image.open(args.label_png))
    if lbl.dtype != np.int32:
        logger.warn('We recomment numpy.int32 for the label, but it has: {}'
                    .format(lbl.dtype))

    logger.info('label shape: {}'.format(lbl.shape))
    logger.info('unique label values: {}'.format(np.unique(lbl)))

    lbl_viz = utils.draw_label(lbl)
    plt.imshow(lbl_viz)
    plt.show()


if __name__ == '__main__':
    main()