🚦 Traffic AI Models Collection
A collection of AI models developed for intelligent traffic management, road safety, and transportation analytics.
📂 Repository Structure
.
├── code/
├── dataset/
├── models/
├── README.md
└── .gitattributes
🤖 Available Models
1. License Plate Detection Model
File: licence_box.pt
This model is trained for automatic license plate detection and recognition from traffic surveillance cameras.
2. Dynamic Traffic Light Model
File: DQN_RL Dynamic traffic.zip
This model analyzes traffic density and helps optimize signal timings dynamically.
3. Traffic Prediction Model
Dataset: west_bengal_traffic_synthetic_data.csv
This model predicts future traffic conditions using historical traffic data.
4. Accident Detection Model
File: accident_best.pt
This model detects road accidents from CCTV or surveillance video streams.
5. Helmet Detection Model
File: helmet_best.pt
This model identifies riders wearing or not wearing helmets.
6. Traffic Light Detection Model
File: light_best.pt
This model detects and classifies traffic lights from camera feeds.
🛠️ Tech Stack
- Python
- YOLO (Ultralytics)
- OpenCV
- NumPy
- PyTorch
