Update utils/preprocessing.py

utils/preprocessing.py

@@ -29,6 +29,8 @@ class ImageProcessor:
             for mask, class_id in zip(results[0].masks.data, results[0].boxes.cls):
                 class_id = int(class_id.item())
                 mask_np = mask.cpu().numpy()
-                # ... [your mask processing logic here] ...
-                segmented[self.class_names[class_id]] = processed_mask
+                mask_resized = cv2.resize(mask_np, (img.shape[1], img.shape[0]), interpolation=cv2.INTER_NEAREST)
+                _, binary_mask = cv2.threshold(mask_resized, 0.5, 255, cv2.THRESH_BINARY)
+                binary_mask = binary_mask.astype(np.uint8)
+                segmented[self.class_names[class_id]] = binary_mask
         return segmented