Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import cv2
|
|
4 |
from deep_sort_realtime.deepsort_tracker import DeepSort
|
5 |
|
6 |
# Initialize YOLO model
|
7 |
-
model = YOLO("
|
8 |
tracker = DeepSort(max_age=30, n_init=3, nn_budget=100)
|
9 |
|
10 |
def count_people_in_video(video_path):
|
|
|
4 |
from deep_sort_realtime.deepsort_tracker import DeepSort
|
5 |
|
6 |
# Initialize YOLO model
|
7 |
+
model = YOLO("setosys_ppl_in_video_small_v1.pt") # Load model
|
8 |
tracker = DeepSort(max_age=30, n_init=3, nn_budget=100)
|
9 |
|
10 |
def count_people_in_video(video_path):
|