Simultaneous-Segmented-Depth-Prediction

Sleeping

Alessio Grancini commited on Feb 14

Commit

8d79b8c

verified ·

1 Parent(s): a2a1782

Update image_segmenter.py

Files changed (1) hide show

image_segmenter.py CHANGED Viewed

@@ -95,7 +95,8 @@ class ImageSegmenter:
                 # object variables
                 (x1, y1, x2, y2) = bounding_boxes[id]
                 center = x1+(x2-x1)//2, y1+(y2-y1)//2
-                objects_data.append([cls, self.model.names[cls], center, self.masks[id], cls_clr])
         return image, objects_data

                 # object variables
                 (x1, y1, x2, y2) = bounding_boxes[id]
                 center = x1+(x2-x1)//2, y1+(y2-y1)//2
+                #objects_data.append([cls, self.model.names[cls], center, self.masks[id], cls_clr])
+                objects_data.append([cls, self.model.names[cls], center, self.masks[id], cls_clr, cls_conf[id]])
         return image, objects_data