Upload detect.py

yolov12/detect.py

@@ -0,0 +1,37 @@
+import subprocess
+import os
+
+def run_yolov12_inference(img_path):
+    project_dir = "static/uploaded"
+    result_dir = os.path.join(project_dir, "yolov12_output")
+
+    command = [
+        "python", "D:/fruit/yolov12/yolo12/yolov12/detect.py",
+        "--weights", "D:/fruit/yolov12/yolo12/yolov12/runs/detect/train7/weights/best.pt",
+        "--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"))
+
+    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
+                    is_damaged = True
+                    break
+    except FileNotFoundError:
+        pass
+
+    return is_damaged