kevin0929 / yolov5s-plus

Yolov5s+

Installation

git clone https://github.com/kevin0929/Yolov5s-plus.git
cd Yolov5s-plus
pip install -r requirements.txt  # install dependencies

Tracking

$ python track.py

Tracking methods

$ python track.py --tracking-method strongsort
                                    ocsort
                                    bytetrack

Tracking sources

Tracking can be run on most video formats

$ python track.py --source 0  # webcam
                           img.jpg  # image
                           vid.mp4  # video
                           path/  # directory
                           path/*.jpg  # glob
                           'https://youtu.be/Zgi9g1ksQHc'  # YouTube
                           'rtsp://example.com/media.mp4'  # RTSP, RTMP, HTTP stream

Filter tracked classes

By default the tracker tracks all MS COCO classes.

If you want to track a subset of the classes that you model predicts, add their corresponding index after the classes flag,

python track.py --source 0 --yolo-weights yolov5s.pt --classes 16 17  # COCO yolov5 model. Track cats and dogs, only