Update draw_label in labelme2voc.py

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

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)