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
最終結果路徑會顯示