安装
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