Update draw_label in labelme2voc.py

finch-lab · Jul 9, 2018 · 65bbb12 · 65bbb12
1 parent 328d50c
commit 65bbb12
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/examples/instance_segmentation/labelme2voc.py b/examples/instance_segmentation/labelme2voc.py
@@ -104,10 +104,8 @@ def main():
                 )
 
             np.save(out_cls_file, cls)
-            label_names = ['%d: %s' % (cls_id, cls_name)
-                           for cls_id, cls_name in enumerate(class_names)]
             clsv = labelme.utils.draw_label(
-                cls, img, label_names, colormap=colormap)
+                cls, img, class_names, colormap=colormap)
             PIL.Image.fromarray(clsv).save(out_clsv_file)
 
             # instance label

diff --git a/examples/semantic_segmentation/labelme2voc.py b/examples/semantic_segmentation/labelme2voc.py
@@ -92,10 +92,8 @@ def main():
 
             np.save(out_lbl_file, lbl)
 
-            label_names = ['%d: %s' % (cls_id, cls_name)
-                           for cls_id, cls_name in enumerate(class_names)]
             viz = labelme.utils.draw_label(
-                lbl, img, label_names, colormap=colormap)
+                lbl, img, class_names, colormap=colormap)
             PIL.Image.fromarray(viz).save(out_viz_file)