bigwiz/new-plate
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
A real-time helmet violation detection system that identifies motorcycles, detects whether riders are wearing helmets, and reads license plates using YOLO, PaddleOCR, and OpenCV.
9 commits 2 branches 0 tags 91 MiB
Find a file
Amit Nandi 1e5119d50a updated readme
2025-03-08 00:17:45 +05:30
input Initial commit 1 2025-03-07 23:48:43 +05:30
models Initial commit 1 2025-03-07 23:48:43 +05:30
.gitignore updated gitignore 2025-03-07 23:59:21 +05:30
LICENSE Initial commit 2025-03-07 23:30:41 +05:30
main.py Initial commit 1 2025-03-07 23:48:43 +05:30
README.md updated readme 2025-03-08 00:17:45 +05:30
requirements.txt Initial commit 1 2025-03-07 23:48:43 +05:30

README.md

🚀 Helmet Violation Detector

A real-time helmet violation detection system that identifies motorcycles, detects whether riders are wearing helmets, and reads license plates using YOLO, PaddleOCR, and OpenCV.

📦 Python 3.12 or below is required for compilation

📌 Features

  • 🏍️ Motorcycle Detection (YOLO-based)
  • 🎩 Helmet Detection (Custom YOLO model)
  • 🚶 Person-Motorcycle Association (using IoU & aspect ratio similarity)
  • 🔍 License Plate Recognition (PaddleOCR)
  • 🚗 License Plate Detection (YOLO-based)
  • 🎥 Real-time Camera Processing (OpenCV)
  • 📊 Report Generation in Excel (OpenPyXL)
  • 🖥️ GUI Interface (PyQt6)

🛠️ Installation

1️⃣ Clone the Repository

git clone https://forge.bitmutex.com/bigwiz/new-plate
cd helmet-violation-detector

2️⃣ Set Up Virtual Environment (Recommended)

python -m venv venv
source venv/bin/activate  # On macOS/Linux
.\venv\Scripts\activate.bat    # On Windows CMD
.\venv\Scripts\activate.ps1    # On Powershell

3️⃣ Install Dependencies

Ensure you have Python 3.8+ installed. Then, run:

pip install -r requirements.txt

📌 Troubleshooting :

  1. If you face PaddleOCR-related dependency issues, make sure to install paddlepaddle:
pip install paddlepaddle
  1. If you face Protobuf related issues, make sure to install correct version of protobuf (4.25.*):
pip install protobuf==4.25.*

🚀 Usage

🔹 GUI Mode

To launch the Helmet Violation Detector GUI, run:

python main.py

🔹 Real-time Camera Detection

Click "Start Real-Time Analysis" in the GUI to process frames from your webcam.

🔹 Processing Images/Videos

  1. Click "Upload Media" and select an image or video.
  2. Click "Analyze Media" to process.
  3. Click "Create Report" to generate an Excel report.
  4. Click "Start Real-Time Mode" to start/stop analysis on connected webcam/camera feed.

📝 Limitations

  • Cannot do excel writes in real time mode , however console outs detected number plates at all times
  • Model Accuracy depends on presented input source quality and fidelity.
  • Currently motorcycle to person association is done however,associations for person to helmet is also required.

⚙️ Dependencies

This project requires:

  • Python 3.8+
  • YOLO (Ultralytics)
  • PaddleOCR
  • OpenCV
  • Mediapipe
  • PyQt6
  • OpenPyXL (Excel support)

You can install them using:

pip install -r requirements.txt