小白YOLOV5训练自己的数据集

假期在家正好学习一下yolov5的使用。

环境配置参考(11条消息) 史上最详细yolov5环境配置搭建+配置所需文件_想到好名再改的博客-CSDN博客_yolov5环境配置以及(11条消息) Yolov5环境配置 配不好来打我_香离的博客-CSDN博客_yolov5环境配置

训练自己的数据集参考了(11条消息) YOLOv5训练自己的数据集详解_tt丫的博客-CSDN博客_yolov5训练自己数据集

但是yawl文件我自己新建了一个。

过程中遇到的问题：

1. UnicodeDecodeError: ‘gbk‘ codec can‘t decode byte 0xaa in position 180: illegal multibyte

参考博客：(11条消息) YOLOv5 报错：UnicodeDecodeError: ‘gbk‘ codec can‘t decode byte 0xaa in position 180: illegal multibyte_业余的炸鸡爱好者的博客-CSDN博客_yolov5报错编码问题以及(11条消息) 记录yolov5出现UnicodeDecodeError中gbk解码问题的解决_Mr.zwX的博客-CSDN博客_yolov5训练模型是报gbk

2. dataset not found

参考博客：(11条消息) yolov5训练时的dataset not found_饮如长鲸吸百川的博客-CSDN博客

下图是我创建的yawl文件内容。

我的yolov5文件夹结构如图：（data1-dial是我做这个项目的数据文件夹）

3. “OSError: [WinError 1455]页面文件太小，无法完成操作。”

参考博客：(11条消息) “OSError: [WinError 1455]页面文件太小，无法完成操作。”解决方案_indigo love的博客-CSDN博客_页面文件太小,无法完成操作

我把值改到了40000才可以正常运行。但是我发现更改这个之后我D盘的空间缩小了很多，之前是130G，现在只有110G了。555~

4、OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.

参考博客：(11条消息) OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized._broad-sky的博客-CSDN博客

我留下的文件位置为D:\Anaconda3\envs\yolov5\Lib\site-packages\torch\lib

5、RuntimeError: CUDA out of memory. Tried to allocate 14.00 MiB

减少banch-size，我改成了8.

6、(19条消息) Pytorch训练出现：RuntimeError: CUDNN_STATUS_INTERNAL_ERROR解决方法_会飞的蜗牛li的博客-CSDN博客

之后成功运行！（epoch=200）

然后使用测试集测试模型：

>python detect.py --weights runs/train/exp15/weights/best.pt --source data1-dial/test/

识别结果如下：（其中一张）

使用测试代码测试模型

python test.py --weights runs/train/exp15/weights/best.pt

得到结果