YoloV5_MCMOT

YoloV5+Sort/DeepSort/ByteTrack/BoTSort/motdt 多类别 尽量简单的调用
tracker的参数从detect_track.py中更改

Install

pip3 install -r requirements.txt
pip3 install cython
pip3 install cython_bbox

Yolov5 + ByteTrack MCMOT 2022/10/2

from detect_track import Detect_Track
model = Detect_Track(tracker='ByteTrack', model_path='./models/yolov5s.pt', 
                    imgsz=(640,640), vis=True)
img_vis, clss, tlwhs, tids = model(img)

Demo

Yolov5 + Sort MCMOT 2022/10/3

from detect_track import Detect_Track
model = Detect_Track(tracker='Sort', model_path='./models/yolov5s.pt', 
                    imgsz=(640,640), vis=True)
img_vis, clss, tlwhs, tids = model(img)  

Demo

Yolov5 + BoTSort MCMOT 2022/10/29

from detect_track import Detect_Track
model = Detect_Track(tracker='BoTSort', model_path='./models/yolov5s.pt', 
                    imgsz=(640,640), vis=True)
img_vis, clss, tlwhs, tids = model(img)  

Yolov5 + DeepSort MCMOT 2022/11/12

from detect_track import Detect_Track
model = Detect_Track(tracker='DeepSort', model_path='./models/yolov5s.pt', 
                    imgsz=(640,640), vis=True)
img_vis, clss, tlwhs, tids = model(img)  

Yolov5 + MotDt MCMOT 2022/11/12

from detect_track import Detect_Track
model = Detect_Track(tracker='motdt', model_path='./models/yolov5s.pt', 
                    imgsz=(640,640), vis=True)
img_vis, clss, tlwhs, tids = model(img)