Răsfoiți Sursa

Fix bug of shapes_to_label for type='instance'

Kentaro Wada 6 ani în urmă
părinte
comite
bc46a933c8
1 a modificat fișierele cu 1 adăugiri și 1 ștergeri
  1. 1 1
      labelme/utils/shape.py

+ 1 - 1
labelme/utils/shape.py

@@ -63,7 +63,7 @@ def shapes_to_label(img_shape, shapes, label_name_to_value, type='class'):
             cls_name = label.split('-')[0]
             if label not in instance_names:
                 instance_names.append(label)
-            ins_id = len(instance_names) - 1
+            ins_id = instance_names.index(label)
         cls_id = label_name_to_value[cls_name]
         mask = shape_to_mask(img_shape[:2], points, shape_type)
         cls[mask] = cls_id