search

ZQPei / deep_sort_pytorch

MOT using deepsort and yolov3 with pytorch
MIT License
2.82k stars 726 forks source link

请问，该项目有转换为多类别多目标跟踪的版本吗？ #173

Open starsky68 opened 4 years ago

starsky68 commented 4 years ago

请问，该项目有转换为多类别多目标跟踪的版本吗？

Single430 commented 3 years ago

这样吗？这个需要修改 yolov3_deepsort.py

# select person class
mask = cls_ids == 0

bbox_xywh = bbox_xywh[mask]
# bbox dilation just in case bbox too small, delete this line if using a better pedestrian detector
bbox_xywh[:, 3:] *= 1.2
cls_conf = cls_conf[mask]