Vehicle Counting using Yolov5 and Deep Sort

On CPU - 12 to 15 FPS

Pre-requisites :

1. Clone the Repository vehicle-counting-yolov5

git clone https://github.com/mahimairaja/vehicle-counting-yolov5.git

cd vehicle-counting-yolov5

2. Clone the legacy Yolo-v5 Repository

git clone https://github.com/ultralytics/yolov5.git

4. Install the libraries

pip install -r requirements.txt

Directory Structure :

After completing the above steps your directory should look like somewhat as of below structure

• vehicle-counting-yolov5
  • deep_sort
  • yolov5
  • input.mp4
  • yolov5s.pt
  • tracker.py
  • requirements.txt

Run the algorithm

python tracker.py 
# This will download model weight - yolov5s.pt to base folder on first execution.

Feel free to conect with me...