Update services/operations_maintenance/pothole_detection.py

services/operations_maintenance/pothole_detection.py

@@ -20,6 +20,7 @@ MODEL_PATH = os.path.abspath(os.path.join(BASE_DIR, "../../models/yolov8m.pt"))
 try:
     model = YOLO(MODEL_PATH)
     logging.info("Loaded YOLOv8m model for pothole detection.")
+    logging.info(f"Model class names: {model.names}")
 except Exception as e:
     logging.error(f"Failed to load YOLOv8m model: {str(e)}")
     model = None
@@ -60,7 +61,7 @@ def process_potholes(frame: np.ndarray) -> Tuple[List[Dict[str, Any]], np.ndarra
     for r in results:
         for box in r.boxes:
             conf = float(box.conf[0])
-            if conf < 0.5:
+            if conf < 0.3:
                 continue
             cls = int(box.cls[0])
             label = model.names[cls]
@@ -77,7 +78,7 @@ def process_potholes(frame: np.ndarray) -> Tuple[List[Dict[str, Any]], np.ndarra
                 "coordinates": [x_min, y_min, x_max, y_max]
             })
 
-            color = (255, 127, 80)  # Coral (red + orange mix)
+            color = (255, 127, 80)  # Coral (red + orange mix) for potholes
             cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), color, 2)
             cv2.putText(
                 frame,