/Vehicle-tracking

Vehicle tracking yolov5 + deepsort

Primary LanguagePython

Vehicle tracking with Yolov5 + Deep Sort with PyTorch

Note

For everyones comes here from youtube or just want to use my repo for inference. This is an very old project just for learning Computer Vision when i was in school. It's been a long time so I forgot almost everything that I did in this toy project. I've editted from this repo and add some lines of code for some constraints. You guys can use that repo for references. Thanks.

Full result video HERE

The detections are generated by YOLOv5 are passed to Deep Sort algorithm which tracks the objects.

Before running the tracker

Python 3.7.12

pip install -r requirements.txt

Config

settings/

config.yml deepsort.yml db_config.yml

Running tracker

cd application/main
python app_track.py

Saving result

Results can be save to databse: upload_db in file config.yml

FastAPI

cd application/main
uvicorn app_API:app --host 0.0.0.0 --port 8000 --reload