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.
4 commits 2 branches 0 tags 91 MiB
Find a file
Amit Nandi 5475f2a214 dcpush
2025-03-07 23:53:51 +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 gitignores 2025-03-07 23:52:26 +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 dcpush 2025-03-07 23:53:51 +05:30
requirements.txt Initial commit 1 2025-03-07 23:48:43 +05:30

README.md

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.

🚀 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.

📌 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://github.com/yourusername/helmet-violation-detector.git
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

🔹 Note:

  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

📌 Troubleshooting

❓ Facing PaddleOCR Issues?

If you get errors related to protobuf or paddleocr, try:

pip install paddlepaddle

or

pip install protobuf==3.20.*