I wrote this project to get familiar with YOLOv5 and also to share and learn from the community.
This project is to rewrite the speculation module in YOLOv5 , so that it can be easily embedded in other projects
- 1、Install the relevant package files
-
pip install -r requirements.txt
- 2、Obtain a weight file: .pt weight file trained by yourself,or get from https://github.com/ultralytics/yolov5/release , to put the file :weights
- 3、Run main.py, modify and apply to your own project according to your needs