ilhamfzri/traffic-violation-detection

Traffic Violation Detection & Recognition is a system was built to detect various traffic violations. Currently, this system only supports detecting running red light violation, helmet violation, and wrong-way.

Traffic Violation Detection & Recognition

---

📝 About The Project

Traffic Violation Detection & Recognition is a system built to detect various traffic violations. Currently, this system only supports detecting running red light violation, helmet violation, and wrong-way. We used YOLOv5 as object detection and SORT as object tracker. If you want to know more about the technical details, please go to the system diagram below.

🚀 Quickstart

1. Clone our repository

   git clone https://github.com/ilhamfzri/traffic-violation-detection.git

2. Install requirements

   cd traffic-violation-detection
   pip install -r requirements.txt

   We recommend you to create an isolated environment using conda.

3. Download our model

   chmod +x download_model.sh
   sh download_model.sh

   We provided the models for vehicle detection and helmet violation.

4. Run the system

   python main.py

   

🔧 System Diagram