Skip to content

Latest commit

 

History

History
58 lines (42 loc) · 1.47 KB

README.md

File metadata and controls

58 lines (42 loc) · 1.47 KB

Wrong-Way-Detection

This project aims to detect vehicles traveling in the wrong direction using computer vision and machine learning techniques.

Project Structure

The repository contains the following key files and directories:

  • data/: Directory containing dataset files
  • model/: Directory for model-related files
  • number-plate-test/: Test scripts for number plate detection
  • wrong-way-vehicle-detection-yolov8/: YOLOv8 implementation for wrong-way vehicle detection
  • detect_wrong.py: Main script for wrong-way detection
  • export.py: Script to export the trained model
  • hubconf.py: PyTorch Hub configuration
  • main.py: Entry point of the application
  • requirements.txt: List of Python dependencies
  • speed.py: Script for vehicle speed estimation
  • test.py: Test script
  • tracker.py: Object tracking implementation
  • train.py: Script for training the model
  • val.py: Validation script
  • visualize.py: Visualization utilities
  • yolov5.pt: Pre-trained YOLOv5 weights file

Setup

  1. Clone the repository:

    git clone https://github.com/KrishGaur1354/Wrong-Way-Detection.git
    
  2. Install the required dependencies:

    pip install -r requirements.txt
    

Usage

To run the wrong-way detection:

python main.py

For training the model:

python train.py

Contributing

Contributions are welcome! Please feel free to submit a Pull Request.

License

This project is licensed under the Unlicense License.