首先我给大家展示一下项目目录

 

 第一步我们需要修改data文件夹下的voc.yaml文件，这里我复制了一份改名为hat.yaml

 需要修改第21，22行的路径，train是图片的训练集，val是图片训练的验证集，nc是标签的数量，names是标签的名称

在models文件夹下有个hub文件夹，里面是你选择的模型，比如我选择的是yolov5s

 这个文件我们只需要修改nc的值和上面的值相同即可，最后修改的就是weights里面的train,py文件

 weights一行是预训练权重文件的路径，该文件从github可以下载

cfg一行就是我上面的yolov5s.yaml文件，我这里换了一个儿名字

data文件就是和上面一样的

epochs是训练的轮数

img-size就是输入图片的分辨率