mod main.py

video_detection.py

@@ -1,7 +1,7 @@
 from ultralytics import YOLO
 import cv2
 
-model = YOLO("models/yolov8m.pt")
+model = YOLO("models/yolo11n_6-2-25.pt")
 
 def draw_boxes(frame, results):
     for r in results:
@@ -41,7 +41,8 @@ def video_detection(cap):
         if not success:
             break
 
-        results = model(frame, stream=True, device='cuda', verbose=False)
+        #results = model(frame, stream=True, device='cuda', verbose=False)
+        results = model(frame, stream=True)
 
         frame = draw_boxes(frame, results)