Update src/features/region_extraction.py

src/features/region_extraction.py

@@ -170,7 +170,18 @@ class RegionExtractor:
             region_image = original_image[y_min:y_max, x_min:x_max].copy()
             region_mask_cropped = region_mask[y_min:y_max, x_min:x_max]
             
+            
             # Critical Fix #4: Region-level confidence aggregation
+            # Ensure region_mask and probability_map have same shape
+            if region_mask.shape != probability_map.shape:
+                import cv2
+                # Resize probability_map to match region_mask
+                probability_map = cv2.resize(
+                    probability_map,
+                    (region_mask.shape[1], region_mask.shape[0]),
+                    interpolation=cv2.INTER_LINEAR
+                )
+            
             region_probs = probability_map[region_mask > 0]
             region_confidence = float(np.mean(region_probs)) if len(region_probs) > 0 else 0.0