An open API service indexing awesome lists of open source software.

https://github.com/krishgaur1354/wrong-way-detection

This project aims to detect vehicles traveling in the wrong direction using computer vision and machine learning techniques.
https://github.com/krishgaur1354/wrong-way-detection

computer-vision python

Last synced: 9 months ago
JSON representation

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

Awesome Lists containing this project

README

          

# Wrong-Way-Detection

This project aims to detect vehicles traveling in the wrong direction using computer vision and machine learning techniques. Published as a research paper at the [15th ICCCNT](https://ieeexplore.ieee.org/abstract/document/10725591).

## 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](LICENSE).