自学内容网 自学内容网

YOLO系列 “.yaml“文件解读

 一、 YOLO系列模型配置的yaml文件解读

        YOLOv5 -v8版本的yaml文件格式基本相同,以下是对yaml文件的解析。

        yaml文件保存了YOLO模型必要的构成信息,包括检测类别数(nc),模型深度(depth _ multiple)、宽度(width_multiple),锚框(anchors大)小、模型的结构(backbone、head)等,如下图中所示。

        ​ 以backbone中第二层的结构[-1, 1, Conv, [128, 3, 2]], # 1-P2/4为例,列表中的第一个参数-1表示该层的输入来自上一层,第二个参数1表示该模块有1个,第三个参数Conv表示该模块为Conv(卷积)模块,第四个参数[128, 3, 2]为当前(Conv卷积)模块的参数,第一个参数表示输出的feature map通道数为128,3表示卷积核为3×3大小,2表示卷积的步长为2。

        如果我们在中间插入或者删减了某一层,则需要修改后面层的来源!


原文地址:https://blog.csdn.net/StopAndGoyyy/article/details/136389804

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!