Spaces:
Running
on
Zero
Running
on
Zero
bacl to yolov8n for pd
Browse files
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("
|
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)
|