/Toll-Gate-Vehicle-Counting

Counting Entrance Vehicle on Toll Gate

Primary LanguagePythonApache License 2.0Apache-2.0

Toll Gate Vehicle Counting

Python PyTorch Template

📼  Demo

demo

📌  Introduction

Toll Gate Vehicle Counting is a system that aims to count the number of vehicles entering toll gates using computer vision technology. The author has blogged about this project which can be seen on ruhyadi.github.io and ruhyadi.medium.io.

The flow diagram of this system can be seen in the image below.

demo

🍿  Inference

python track.py \
    --source ./demo/toll_gate.mp4 \
    --yolo-weights ./weights/yolov5s.pt \
    --strong-sort-weights ./weights/osnet_x0_25_msmt17.pt \
    --classes 2 5 7 \
    --save-vid

❤️ Acknowledgement

The project is part of Nodeflux Intership x Kampus Merdeka