Update image_segmenter.py

image_segmenter.py

@@ -40,7 +40,9 @@ 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)
 
         cls_ids = predictions[0].boxes.cls.cpu().numpy()
         bounding_boxes = predictions[0].boxes.xyxy.int().cpu().numpy()