Spaces:
Running
on
Zero
Running
on
Zero
File size: 510 Bytes
9e56ba5 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from .strong_sort import StrongSORT
__all__ = ['StrongSORT', 'build_tracker']
def build_tracker(cfg, use_cuda):
return StrongSORT(cfg.STRONGSORT.REID_CKPT,
max_dist=cfg.STRONGSORT.MAX_DIST, min_confidence=cfg.STRONGSORT.MIN_CONFIDENCE,
nms_max_overlap=cfg.STRONGSORT.NMS_MAX_OVERLAP, max_iou_distance=cfg.STRONGSORT.MAX_IOU_DISTANCE,
max_age=cfg.STRONGSORT.MAX_AGE, n_init=cfg.STRONGSORT.N_INIT, nn_budget=cfg.STRONGSORT.NN_BUDGET, use_cuda=use_cuda)
|