labelmeWithAI/examples/tutorial/load_label_png.py

#!/usr/bin/env python

from __future__ import print_function

import os.path as osp

import numpy as np
import PIL.Image


here = osp.dirname(osp.abspath(__file__))


def main():
    label_png = osp.join(here, 'apc2016_obj3_json/label.png')
    print('Loading:', label_png)
    print()

    lbl = np.asarray(PIL.Image.open(label_png))
    labels = np.unique(lbl)

    label_names_txt = osp.join(here, 'apc2016_obj3_json/label_names.txt')
    label_names = [name.strip() for name in open(label_names_txt)]
    print('# of labels:', len(labels))
    print('# of label_names:', len(label_names))
    if len(labels) != len(label_names):
        print('Number of unique labels and label_names must be same.')
        quit(1)
    print()

    print('label: label_name')
    for label, label_name in zip(labels, label_names):
        print('%d: %s' % (label, label_name))


if __name__ == '__main__':
    main()