- Notifications
You must be signed in to change notification settings - Fork13
Fast MOT base on yolo+deepsort, support yolo3 and yolo4
License
NotificationsYou must be signed in to change notification settings
GlassyWing/yolo_deepsort
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
MOT base on yolo3/yolo4+deepsort, different from the official use of numpy to implement sort, the sort here reimplemented with pytorch, so it running at GPU.
pytorch >= 1.3torchvision >= 0.4.0opencv-python >= 4.1- Clone the repositorty
git clone https://github.com/GlassyWing/yolo_deepsort
- Download weights of yolo3 or yolo4
cd weights/wget https://pjreddie.com/media/files/yolov3.weightswget https://pjreddie.com/media/files/yolov3-tiny.weightswget https://drive.google.com/file/d/1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT/view- Download weights of DeepSort
# download ckpt.t7 fromhttps://drive.google.com/drive/folders/1xhG0kRH1EX5B9_Iz8gQJb7UNnn_riXi6 to this folder- Run example
python video_deepsort.py
If you do not want to run the tracker, set the parametertracker toNone:
video_detector = VideoDetector(... tracker=None)Please read thetutorial.
This library does not contain a feasible training program, please refer to the training: