This project aims to detect vehicles traveling in the wrong direction using computer vision and machine learning techniques.
The repository contains the following key files and directories:
data/
: Directory containing dataset filesmodel/
: Directory for model-related filesnumber-plate-test/
: Test scripts for number plate detectionwrong-way-vehicle-detection-yolov8/
: YOLOv8 implementation for wrong-way vehicle detectiondetect_wrong.py
: Main script for wrong-way detectionexport.py
: Script to export the trained modelhubconf.py
: PyTorch Hub configurationmain.py
: Entry point of the applicationrequirements.txt
: List of Python dependenciesspeed.py
: Script for vehicle speed estimationtest.py
: Test scripttracker.py
: Object tracking implementationtrain.py
: Script for training the modelval.py
: Validation scriptvisualize.py
: Visualization utilitiesyolov5.pt
: Pre-trained YOLOv5 weights file
-
Clone the repository:
git clone https://github.com/KrishGaur1354/Wrong-Way-Detection.git
-
Install the required dependencies:
pip install -r requirements.txt
To run the wrong-way detection:
python main.py
For training the model:
python train.py
Contributions are welcome! Please feel free to submit a Pull Request.
This project is licensed under the Unlicense License.