Luigi commited on
Commit
48d17a2
·
1 Parent(s): 7a13da5

bacl to yolov8n for pd

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ def process_video(video_file, threshold_secs):
28
  out = cv2.VideoWriter(out_path, fourcc, fps, (width, height))
29
 
30
  # Load the YOLOv8 model for person detection (medium model for better accuracy)
31
- yolo_model = YOLO("yolov8m.pt")
32
 
33
  # Initialize DeepSORT tracker
34
  tracker = DeepSort(max_age=30, n_init=3, embedder="mobilenet", half=True)
 
28
  out = cv2.VideoWriter(out_path, fourcc, fps, (width, height))
29
 
30
  # Load the YOLOv8 model for person detection (medium model for better accuracy)
31
+ yolo_model = YOLO("yolov8n.pt")
32
 
33
  # Initialize DeepSORT tracker
34
  tracker = DeepSort(max_age=30, n_init=3, embedder="mobilenet", half=True)