Explorar o código

Fix segmentation generation for rectangles

Shannon %!s(int64=5) %!d(string=hai) anos
pai
achega
ecf6362a37
Modificáronse 1 ficheiros con 5 adicións e 0 borrados
  1. 5 0
      examples/instance_segmentation/labelme2coco.py

+ 5 - 0
examples/instance_segmentation/labelme2coco.py

@@ -126,6 +126,11 @@ def main():
                 masks[instance] = mask
 
             points = np.asarray(points).flatten().tolist()
+            
+            if shape['shape_type'] == 'rectangle':
+                x_1, y_1, x_2, y_2 = points
+                points = [x_1, y_1, x_2, y_1, x_2, y_2, x_1, y_2]
+            
             segmentations[instance].append(points)
         segmentations = dict(segmentations)