Update image_segmenter.py

image_segmenter.py

@@ -40,10 +40,10 @@ class ImageSegmenter:
         # params
         objects_data = [] 
         image = image.copy()
-        #predictions = self.model.predict(image)
+        predictions = self.model.predict(image)
         # Use the original image resolution (1512x1008) instead of default YOLO resize
         # predictions = self.model.predict(image, imgsz=(1512, 1008), conf=self.confidence_threshold)
-        predictions = self.model.predict(image, imgsz=(1536, 1024), conf=self.confidence_threshold)
+        # predictions = self.model.predict(image, imgsz=(1536, 1024), conf=self.confidence_threshold)
 
         cls_ids = predictions[0].boxes.cls.cpu().numpy()
         bounding_boxes = predictions[0].boxes.xyxy.int().cpu().numpy()