Create detection_service.py

services/detection_service.py

@@ -0,0 +1,29 @@
+import torch
+from models.solar_model import load_solar_model
+from models.windmill_model import load_windmill_model
+
+def detect_faults_solar(model, image):
+    results = model(image)
+    faults = []
+
+    for detection in results.xywh[0]:  # Assuming the results are in this format
+        class_id = int(detection[5].item())  # Get the class ID of the detected object
+        if class_id == 0:  # Cracks
+            faults.append({"type": "Crack", "location": (detection[0], detection[1])})
+        elif class_id == 1:  # Damage
+            faults.append({"type": "Damage", "location": (detection[0], detection[1])})
+
+    return faults
+
+def detect_faults_windmill(model, image):
+    results = model(image)
+    faults = []
+
+    for detection in results.xywh[0]:  # Assuming the results are in this format
+        class_id = int(detection[5].item())  # Get the class ID of the detected object
+        if class_id == 2:  # Blade damage
+            faults.append({"type": "Blade Damage", "location": (detection[0], detection[1])})
+        elif class_id == 3:  # Motor fault
+            faults.append({"type": "Motor Fault", "location": (detection[0], detection[1])})
+
+    return faults