三分钟学会使用系列(YOLOv5)|YOLOv5 yaml文件解读!
本文旨在让读者快速了解相关技术并运用,欢迎关注。
1. YOLOv5中的yaml配置文件解读
yaml文件保存了YOLOv5模型的必要信息,包括检测类别数(nc),模型深度(depth_multiple)、宽度(width_multiple),锚框(anchors大)小、模型的结构(backbone、head)等,如图1中所示。
以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,如图1。