1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
#!/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()
|
