Update yolov12/yolo.py

yolov12/yolo.py

@@ -1,37 +1,32 @@
-import subprocess
 import os
+import subprocess
 
 def run_yolov12_inference(img_path):
     project_dir = "static/uploaded"
     result_dir = os.path.join(project_dir, "yolov12_output")
+    label_file = os.path.join(result_dir, "labels", os.path.basename(img_path).replace(".jpg", ".txt"))
 
     command = [
-        "python", "../yolov12/yolo.py",
-        "--weights", "../runs/detect/train7/weights/best.pt",
+        "python3", "yolov12/detector.py",  # Simulated CLI
         "--source", img_path,
-        "--conf", "0.25",
-        "--save-txt",
-        "--save-conf",
         "--project", project_dir,
         "--name", "yolov12_output",
-        "--exist-ok"
     ]
 
-    subprocess.run(command, check=True)
-
-    # Try reading the prediction txt file
-    file_name = os.path.basename(img_path)
-    label_file = os.path.join(result_dir, "labels", file_name.replace(".jpg", ".txt"))
+    try:
+        subprocess.run(command, check=True)
+    except Exception as e:
+        print("❌ Subprocess failed:", e)
 
     is_damaged = False
     try:
         with open(label_file, "r") as f:
             for line in f:
                 class_id = int(line.split()[0])
-                if class_id == 0:  # assume damaged_apple is class 0
+                if class_id == 0:
                     is_damaged = True
                     break
     except FileNotFoundError:
-        pass
+        print("⚠️ No label file found:", label_file)
 
     return is_damaged