Update services/crack_detection_service.py

services/crack_detection_service.py

@@ -8,11 +8,11 @@ def detect_cracks_and_holes(frame):
         blurred = cv2.GaussianBlur(gray, (5, 5), 0)
 
         # Edge detection using Canny
-        edges = cv2.Canny(blurred, 50, 150)
+        edges = cv2.Canny(blurred, 30, 100)  # Lower thresholds to detect more cracks
 
         # Morphological operations to enhance cracks and holes
         kernel = np.ones((3, 3), np.uint8)
-        dilated = cv2.dilate(edges, kernel, iterations=1)
+        dilated = cv2.dilate(edges, kernel, iterations=2)
         eroded = cv2.erode(dilated, kernel, iterations=1)
 
         # Find contours
@@ -22,7 +22,7 @@ def detect_cracks_and_holes(frame):
         for contour in contours:
             # Filter small contours (noise)
             area = cv2.contourArea(contour)
-            if area < 50:
+            if area < 30:  # Reduced threshold to catch smaller cracks
                 continue
 
             # Get bounding box
@@ -35,22 +35,22 @@ def detect_cracks_and_holes(frame):
             circularity = 4 * np.pi * area / (perimeter * perimeter) if perimeter > 0 else 0
 
             # Classify as crack or hole
-            if 0.1 < aspect_ratio < 10 and circularity < 0.5:  # Long, thin shapes are cracks
+            if 0.05 < aspect_ratio < 20 and circularity < 0.4:  # Adjusted for more crack detection
                 item_type = 'crack'
                 # Check if it's an underlying crack (longer and thinner)
-                if aspect_ratio > 5:
+                if aspect_ratio > 8:
                     severity = 'Underlying'
-                elif area > 5000:
+                elif area > 4000:
                     severity = 'Severe'
-                elif area > 1000:
+                elif area > 800:
                     severity = 'Moderate'
                 else:
                     severity = 'Minor'
-            elif circularity > 0.7:  # Circular shapes are holes
+            elif circularity > 0.6:  # Circular shapes are holes
                 item_type = 'hole'
-                if area > 5000:
+                if area > 4000:
                     severity = 'Severe'
-                elif area > 1000:
+                elif area > 800:
                     severity = 'Moderate'
                 else:
                     severity = 'Minor'