Update services/operations_maintenance/crack_detection.py

services/operations_maintenance/crack_detection.py

@@ -6,7 +6,7 @@ import random
 
 # Load YOLOv8m-seg model for crack detection
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-MODEL_PATH = os.path.join(BASE_DIR, "../../models/yolov8m-seg.pt")
+MODEL_PATH = os.path.abspath(os.path.join(BASE_DIR, "../../models/yolov8m-seg.pt"))
 model = YOLO(MODEL_PATH)
 
 def detect_cracks_and_objects(frame):
@@ -21,7 +21,6 @@ def detect_cracks_and_objects(frame):
     results = model(frame)
 
     detected_items = []
-    line_counter = 1  # Initialize counter for numbered labels
 
     # Process detections
     for r in results:
@@ -41,11 +40,8 @@ def detect_cracks_and_objects(frame):
             if label == "crack":
                 severity = random.choice(["low", "medium", "high"])
             
-            # Add numbered label
-            detection_label = f"Line {line_counter} - {label.capitalize()} (Conf: {conf:.2f})"
             item = {
                 "type": label,
-                "label": detection_label,
                 "confidence": conf,
                 "coordinates": [x_min, y_min, x_max, y_max]
             }
@@ -54,6 +50,4 @@ def detect_cracks_and_objects(frame):
 
             detected_items.append(item)
 
-            line_counter += 1
-
     return detected_items