Переглянути джерело

Show label_value in draw_label

Naoya Yamaguchi 6 роки тому
батько
коміт
683a049450
1 змінених файлів з 2 додано та 1 видалено
  1. 2 1
      labelme/utils.py

+ 2 - 1
labelme/utils.py

@@ -117,7 +117,8 @@ def draw_label(label, img=None, label_names=None, colormap=None):
         fc = colormap[label_value]
         p = plt.Rectangle((0, 0), 1, 1, fc=fc)
         plt_handlers.append(p)
-        plt_titles.append(label_name)
+        plt_titles.append('{value}: {name}'
+                          .format(value=label_value, name=label_name))
     plt.legend(plt_handlers, plt_titles, loc='lower right', framealpha=.5)
 
     f = io.BytesIO()