Yolov5 + StrongSORT with OSNet
- Clone this project
- Download all files from LINK TO DRIVE
- Copy and paste any files from .zip to root folder
- Create conda environment
$ conda env create -f environment.yml
- Or Install requirements.txt
$ pip install -r requirements.txt
You can change weight.pt by put a new weight in weight folder and change a parameter in object_tracker.py
- YOLO_WEIGHTS_PATH = WEIGHTS / 'custom-weight.pt'
- STRONG_SORT_WEIGHTS = WEIGHTS / 'custom-strong_sort-weight'