This project is an implementation of YOLOv5, a popular deep learning algorithm for real-time object detection, using Python. YOLOv5 is a state-of-the-art object detection model that achieves high accuracy and fast inference speed.
- Python 3.x
- PyTorch
- OpenCV
- NumPy
You can install the required Python packages using pip:
pip install -r requirements.txt
- Create venv
python -m venv env
- Use env
source env/bin/activate
- Preprocessing data
python pre_processing.py
- Train model
python train.py
- Predict
python predict.py
- YOLOv5 GitHub repository: https://github.com/ultralytics/yolov5
- YOLOv5 paper: YOLOv5: A Universal Object Detector
This project is licensed under the MIT License - see the LICENSE file for details.