关于yolov5训练需要更改的参数汇总
首先我给大家展示一下项目目录
第一步我们需要修改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就是输入图片的分辨率
原文地址：https://blog.csdn.net/2301_76846375/article/details/142486109
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！