Skip to content
Snippets Groups Projects
Commit 65acf304 authored by Kentaro Wada's avatar Kentaro Wada
Browse files

Show class_id in Visualization

parent e70bcaeb
Branches
No related tags found
No related merge requests found
......@@ -89,8 +89,10 @@ def main():
# lbl_pil.putpalette((colormap * 255).flatten())
lbl_pil.save(out_lbl_file)
label_names = ['%d: %s' % (cls_id, cls_name)
for cls_id, cls_name in enumerate(class_names)]
viz = labelme.utils.draw_label(
lbl, img, class_names, colormap=colormap)
lbl, img, label_names, colormap=colormap)
PIL.Image.fromarray(viz).save(out_viz_file)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment