YOLOv5安装、部署、训练自己数据集模型、测试

安装

1 安装依赖环境

1.1 从Github上clone仓库

git clone https://github.com/ultralytics/yolov5 

1.2 使用conda建立虚拟环境

conda create -n yolov5 python=3.8 
conda activate yolov5

1.3 进入仓库所在目录 安装相关依赖包

cd yolov5
pip3 install -r requirements.txt

可能会提示报错需要安装scikit-build

pip3 install scikit-build

2 下载模型测试

2.1 测试

python3 detect.py --source data/images/ --weights weights/yolov5s.pt --conf 0.4

2.2 训练

python3 train.py --img 640 --batch 16 --epochs 200 --data ./data/coco128.yaml --cfg ./models/yolov5s.yaml --weights ./weights/yolov5s.pt --device 0

2.3 测试

python3 detect.py --source ./inference/input/ --weight ./runs/train/exp/weights/best.pt