Spaces:
Sleeping
Sleeping
Update Yolov5_Deepsort/tracker.py
Browse files
Yolov5_Deepsort/tracker.py
CHANGED
@@ -7,7 +7,7 @@ import cv2
|
|
7 |
|
8 |
palette = (2 ** 11 - 1, 2 ** 15 - 1, 2 ** 20 - 1)
|
9 |
cfg = get_config()
|
10 |
-
cfg.merge_from_file("deep_sort/configs/deep_sort.yaml")
|
11 |
deepsort = DeepSort(cfg.DEEPSORT.REID_CKPT,
|
12 |
max_dist=cfg.DEEPSORT.MAX_DIST, min_confidence=cfg.DEEPSORT.MIN_CONFIDENCE,
|
13 |
nms_max_overlap=cfg.DEEPSORT.NMS_MAX_OVERLAP, max_iou_distance=cfg.DEEPSORT.MAX_IOU_DISTANCE,
|
|
|
7 |
|
8 |
palette = (2 ** 11 - 1, 2 ** 15 - 1, 2 ** 20 - 1)
|
9 |
cfg = get_config()
|
10 |
+
cfg.merge_from_file("Yolov5_Deepsort/deep_sort/configs/deep_sort.yaml")
|
11 |
deepsort = DeepSort(cfg.DEEPSORT.REID_CKPT,
|
12 |
max_dist=cfg.DEEPSORT.MAX_DIST, min_confidence=cfg.DEEPSORT.MIN_CONFIDENCE,
|
13 |
nms_max_overlap=cfg.DEEPSORT.NMS_MAX_OVERLAP, max_iou_distance=cfg.DEEPSORT.MAX_IOU_DISTANCE,
|