使用YOLO5训练数据集

下载模型 

在github下载官方的yolov5模型,进入网站直接搜索yolo第一个就是，点进去选择对应版本号后，download zip即可获得原始的一个模型

下载官方所给预训练模型

点击v5.0 release,进入后下滑至最底，下载需要的预训练权重。 个人使用yolov5s进行行人检测

 

训练 

解压缩下载的yolov5包，在pycharm中打开，修改配置文件

data文件夹 

首先是yolov5-5.0\data,将voc.yaml复制一份并重命名，我重命名为person.yaml

然后修改这个文件修改参数

修改前：

修改后：

修改了四个部分，

1：注销第十行；

2：修改13和14行，改为自己存储训练集和测试集的绝对地址；

3：修改17行，数字代表需要检测得到类别数；

4：修改20行，20行代表要检测类别的名称。

models文件夹

修改yolov5-5.0\models，将yolov5s.yaml复制一份并重命名，我重命名为person_yolov5s.yaml

打开重命名的文件，将第二行nc=80修改为nc=1，表识别一类。

找到train.py文件，修改前三个参数

464行修改成我们下载的训练好的yolov5s模型权重

465行修改成model文件下我们新命名的文件地址

465行修改成data文件下我们新命名的文件地址

第468行修改训练轮次

到这就可以进行训练了，结果会在yolov5-5.0\runs\train下产生权重文件存储最好的和最后一次的训练数据

推理

找到detect.py文件进行推理

修改151行为训练得到的权重地址

152行是传入的图片或视频地址来进行推理

结果存储在yolov5-5.0\runs\detect下