Update services/crack_detection_service.py

services/crack_detection_service.py

@@ -4,10 +4,10 @@ import numpy as np
 # Define road boundary (ROI) for detection
 ROAD_BOUNDARY = (50, 50, 590, 430)  # Example: x_min, y_min, x_max, y_max (adjust accordingly)
 
-def detect_cracks(frame):
+def detect_cracks_and_objects(frame):
     """
-    Detects cracks in the given video frame.
-    Uses edge detection and contour analysis to detect linear crack shapes.
+    Detects cracks and objects (potholes) in the given video frame.
+    Uses edge detection and contour analysis to detect cracks and objects.
     """
     try:
         # Convert frame to grayscale for crack detection
@@ -42,7 +42,12 @@ def detect_cracks(frame):
                         'confidence': 0.95  # Simulated confidence
                     })
 
-        return cracks
+        # Detecting other objects (e.g., potholes) in the frame can be added here
+        objects = []  # Placeholder for detected objects
+        # Object detection logic using custom or pretrained models (e.g., DETR)
+        # You can add object detection for potholes here.
+
+        return cracks + objects
     except Exception as e:
-        print(f"Error in crack detection: {str(e)}")
+        print(f"Error in detection: {str(e)}")
         return []