Update services/road_safety/pothole_crack_detection.py

services/road_safety/pothole_crack_detection.py

@@ -4,7 +4,7 @@ from ultralytics import YOLO
 from typing import List, Tuple, Dict, Any
 
 # Load YOLOv8 model for pothole and crack detection
-model = YOLO("models/yolov8n.pt")  # Use lighter model for faster inference
+model = YOLO("models/yolov8n.pt")
 
 def detect_potholes_and_cracks(frame: np.ndarray) -> Tuple[List[Dict[str, Any]], np.ndarray]:
     """
@@ -15,7 +15,7 @@ def detect_potholes_and_cracks(frame: np.ndarray) -> Tuple[List[Dict[str, Any]],
         Tuple of (list of detections, annotated frame).
     """
     # Perform inference
-    results = model(frame, classes=[0, 1], conf=0.5)  # Assuming class 0 is pothole, class 1 is crack
+    results = model(frame, classes=[0, 1], conf=0.5)  # Class 0: pothole, Class 1: crack
     
     detections = []
     for i, r in enumerate(results[0].boxes):