Alessio Grancini commited on
Commit
1d9eab5
·
verified ·
1 Parent(s): 25c991c

Update image_segmenter.py

Browse files
Files changed (1) hide show
  1. 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