Updated Scripts

inference/detect_image.py

@@ -10,7 +10,7 @@ if len(sys.argv) < 2:
 image_path = sys.argv[1]
 
 # Load trained model
-model = YOLO("runs/detect/weights/best.pt")
+model = YOLO("best50.pt")
 
 # Read image
 frame = cv2.imread(image_path)

inference/detect_video.py

@@ -1,19 +1,36 @@
 from ultralytics import YOLO
 import cv2
+import sys
 
-model = YOLO("runs/detect/weights/best.pt")
-video_path = "test_video.mp4"
+# Check for command-line argument
+if len(sys.argv) < 2:
+    print("Usage: python detect_video.py <video_path>")
+    exit()
 
+video_path = sys.argv[1]
+
+model = YOLO("best50.pt")
 cap = cv2.VideoCapture(video_path)
-while cap.isOpened():
+
+if not cap.isOpened():
+    print(f"Error: Could not open video file {video_path}.")
+    exit()
+
+print("Press 'q' to quit.")
+while True:
     ret, frame = cap.read()
-    if not ret: break
+    if not ret:
+        print("End of video or failed to grab frame.")
+        break
 
-    results = model(frame, conf=0.5, imgsz=640, augment=False)
+    results = model(frame, conf=0.5, iou=0.4, imgsz=640, augment=False)
     annotated_frame = results[0].plot()
 
     cv2.imshow("Video Detection", annotated_frame)
-    if cv2.waitKey(1) == ord('q'): break
+
+    if cv2.waitKey(1) & 0xFF == ord('q'):
+        break
 
 cap.release()
-cv2.destroyAllWindows()
+cv2.destroyAllWindows()
+print("Video closed.")

inference/detect_webcam.py

@@ -1,18 +1,28 @@
 from ultralytics import YOLO
 import cv2
 
-model = YOLO("runs/detect/weights/best.pt")
-
+model = YOLO("best50.pt")
 cap = cv2.VideoCapture(0)  # 0 = default webcam
-while cap.isOpened():
+
+if not cap.isOpened():
+    print("Error: Could not open webcam.")
+    exit()
+
+print("Press 'q' to quit.")
+while True:
     ret, frame = cap.read()
-    if not ret: break
+    if not ret:
+        print("Failed to grab frame.")
+        break
 
-    results = model(frame, conf=0.5, imgsz=640, augment=False)
+    results = model(frame, conf=0.5, iou=0.4, imgsz=640, augment=False)
     annotated_frame = results[0].plot()
 
     cv2.imshow("Webcam Detection", annotated_frame)
-    if cv2.waitKey(1) == ord('q'): break
+
+    if cv2.waitKey(1) & 0xFF == ord('q'):
+        break
 
 cap.release()
-cv2.destroyAllWindows()
+cv2.destroyAllWindows()
+print("Webcam closed.")