Idk

rpi/board-detector/debug.py

@@ -0,0 +1,31 @@
+#!/usr/bin/env python3
+import cv2
+import numpy as np
+from ultralytics import YOLO
+from paths import *
+
+
+def main():
+    model = YOLO(model_path)
+
+    # Load image
+    image = cv2.imread(img_path)
+    if image is None:
+        print(f"Failed to read {img_path}")
+        return
+
+    height, width = image.shape[:2]
+
+    warped = image  # For now assume top-down view
+
+    # Run YOLO detection
+    results = model(warped)
+    res = results[0]
+
+    debug_img = res.plot()  # draws boxes around detected objects
+    cv2.imshow("Detections", debug_img)
+    cv2.waitKey(0)
+    cv2.destroyAllWindows()
+
+if __name__ == "__main__":
+    main()