Update image_segmenter.py

image_segmenter.py

@@ -42,7 +42,8 @@ class ImageSegmenter:
         image = image.copy()
         #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=(1512, 1008), 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()