Yolov5如何训练自定义的数据集，以及使用GPU训练，涵盖报错解决

本文主要讲述了Yolov5如何训练自定义的数据集，以及使用GPU训练，涵盖报错解决，案例是检测图片中是否有救生圈。 最后的效果图大致如下：

效果图1	效果图2

前言

系列文章

1、详细讲述Yolov5从下载、配置及如何使用GPU运行

2、Labelimg标注自己的数据集，及如何划分训练集和验证集，应用于Yolov5

上一篇文章中，已经介绍了该如何标注自己的数据集，以及该如何给他们分类，接下来的话，就是根据我们已经标注好的数据集来进行训练啦。

将我们之前划分好的数据集放入项目中：

补充：通过文件夹将数据集复制放进去的，pycharm可以索引的更快，不然会卡很久。

一、修改配置文件

我们需要指定数据集的位置，首先就是要修改 coco128.yaml 配置文件

1.1、修改 coco128.yaml 配置文件

我们复制data 文件夹下 coco128.yaml 配置文件，并重命名为 blog_demo.yaml

1.2、修改 yolov5s.yaml 配置文件

我们复制 models 文件夹下 yolov5s.yaml 配置文件，并重命名为 yolov5s_blog.yaml

参数的细节可能需要各位朋友自己去了解啦，我这里只是将类别数目修改成和data/blog_demo.yaml文件对应。

二、了解 train.py 参数

此处暂时不做深究，我们当前的任务是跑通整个代码。

主要是了解截图里面部分参数，这主要是因为训练的时候，不同的机器硬件参数不同，能做的事情也不同，所以相应需要调整，尽可能的把性能发挥到极致。