yolov5安装教程

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

最终结果路径会显示


如果遇到什么不懂的地方直接关注公众号留言(本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须在文章页面给出原文连接,否则保留追究法律责任的权利。)
作者:newmiracle
出处:https://www.cnblogs.com/newmiracle/

 
原文地址:https://www.cnblogs.com/newmiracle/p/15322880.html