# Traffic light detection

**Repository Path**: nintendo2022/traffic-light-detection

## Basic Information

- **Project Name**: Traffic light detection
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2023-11-27
- **Last Updated**: 2023-12-21

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

<!-- # TrafficViolationDetection


<h1>
<p align="center">
  <img src="img/Full Diagram-01.png" alt="Logo" height=400>
</h1>

 -->


<!-- LOGO -->

<br />
<h1>
<p align="center">
  <img src="img/tvdr.png" alt="Logo" height=250>
  <br>Traffic Violation Detection & Recognition
</h1>
<p align="center">
<a href="https://www.python.org/"><img alt="Actions Status" src="https://img.shields.io/badge/Made%20with-Python-1f425f.svg"></a>
<a href="https://github.com/psf/black"><img alt="Code style: black" src="https://img.shields.io/badge/code%20style-black-000000.svg"></a>
<!-- <a href="https://colab.research.google.com/drive/1bwnepojzg_1T56IWPee_n998nEh7catM?usp=sharing"><img alt="Actions Status" src="https://colab.research.google.com/assets/colab-badge.svg"></a> -->
</p>


</p>

-----

## 📝 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
    ```bash
    git clone https://github.com/ilhamfzri/traffic-violation-detection.git
    ```

2. Install requirements
    ```bash
    cd traffic-violation-detection
    pip install -r requirements.txt
    ```
    We recommend you to create an isolated environment using conda.

3. Download our model
      ```bash
    chmod +x download_model.sh
    sh download_model.sh
    ```
    We provided the models for vehicle detection and helmet violation.

4. Run the system
      ```bash
      python main.py
      ```
    <h1>
    <p align="center">
      <img src="img/example.png" alt="Logo" height=400>
    </h1>



##  🔧 System Diagram
<h1>
<p align="center">
  <img src="img/Full Diagram-01.png" alt="Logo" height=400>
</h1>