Alessio Grancini
commited on
Update image_segmenter.py
Browse files- image_segmenter.py +1 -0
image_segmenter.py
CHANGED
@@ -98,6 +98,7 @@ class ImageSegmenter:
|
|
98 |
# object variables
|
99 |
(x1, y1, x2, y2) = bounding_boxes[id]
|
100 |
center = x1+(x2-x1)//2, y1+(y2-y1)//2
|
|
|
101 |
#objects_data.append([cls, self.model.names[cls], center, self.masks[id], cls_clr])
|
102 |
objects_data.append([cls, self.model.names[cls], center, self.masks[id], cls_clr, cls_conf[id]])
|
103 |
|
|
|
98 |
# object variables
|
99 |
(x1, y1, x2, y2) = bounding_boxes[id]
|
100 |
center = x1+(x2-x1)//2, y1+(y2-y1)//2
|
101 |
+
print(f"Center Coordinates: {center}")
|
102 |
#objects_data.append([cls, self.model.names[cls], center, self.masks[id], cls_clr])
|
103 |
objects_data.append([cls, self.model.names[cls], center, self.masks[id], cls_clr, cls_conf[id]])
|
104 |
|