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:\yolo\yoloa\train\images
val: I:\yolo\yoloa\valid\images

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

最终结果路径会显示