yolov5安装教程
在anaconda环境下安装
新建个 yolov5文件夹 先下个yolov5 https://github.com/ultralytics/yolov5
ps:必须是英文路径 不能中文路径不然训练的时候识别不了
然后
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
然后下个数据集 别用官方的coc2018 会直接卡死 自己训练个 labelimg标签制作可以参照下面教程 labelimg直接去github下个win版本就行了 注意必须英文路径
https://blog.csdn.net/qq_45701791/article/details/113992622 主要一定要切换到yolo模式下 别用voc模式 生成的是txt
里面数据集yaml路径要注意下 写绝对肯定行
train: I:yoloyoloa rainimages
val: I:yoloyoloavalidimages
nc: 1
names: ['a']
以下是我自己制作的数据集可以参考下
链接:https://pan.baidu.com/s/1Wj8K_luMQtDL7Whkosnmzw
提取码:6zz2
--来自百度网盘超级会员V3的分享
以下先生成下yolov5s.pt 训练模型要用
python detect.py --weights yolov5s.pt --img 640 --conf 0.25 --source data/images/
以下命令是训练
python train.py --img 640 --batch 50 --epochs 100 --data ../yoloa/a.yaml --weights yolov5s.pt --nosave --cache
会生成个best.pt 下面要用
接下来就是验证下准确度
detect.py --weights ./runs/train/exp39/weights/best.pt --img 640 --conf 0.25 --source ../yoloa/test/images
最终结果路径会显示