Yolov5训练制定数据集-完整版
Yolov5训练制定数据集
1 创建数据集
1.1 创建如下⽂件夹
1.2 标注数据
将结果按照images和Labels放到1.1中对应的⽂件中
1.3 创建custom_data.yaml⽂件
图片下载站源码
到1 ，2 ，3步骤，到data⽂件夹下的coco128.yaml，复制⼀份，改名为custom_data.yaml，将custom_data.yaml中添加3步骤的内容。
2 训练
python train.py --img 640 --batch 2 --epochs 100 --data custom_data.yaml --weights yolov5s.pt --nosave --cache
3 训练过程遇到的问题
3.1 可以会出现 “OSError: [WinError 1455] 页⾯⽂件太⼩”。
原因:python所在磁盘的虚拟内存不够。
解决⽅案：
（⼀）⽅法⼀：
将yolov5中utils⽂件下dataset.py的num_workers修改为0，如下图所⽰：
(2）给python所在盘分配虚拟内存
“查看⾼级系统配置”
3.2 opencv-python>=
4.1.2 not found
pip install opencv-python ⾃动下载最新版本的
3.s.CommError: check_hostname requires server_hostname # ⽅框注册，然后会获得⼀个key
wandb init
解决⽅案 ： 科学上⽹关掉
3.3 Can’t get attribute ‘SPPF’ on <module ‘modelsmon’>
解决⽅案：⽂件权重问题，使⽤程序原有的权重⽂件。