YOLOV8的yolov8.yaml解读

无影663

于 2024-08-29 18:31:44 发布

阅读量285

点赞数 1

文章标签： YOLO

本文链接：https://blog.csdn.net/qq_66919612/article/details/141685510

版权

yaml文件保存了YOLO模型必要的构成信息，包括检测类别数（nc），模型深度（depth _ multiple）、宽度（width_multiple），锚框（anchors大）小、模型的结构（backbone、head）等，通过yaml文件我们可以推测出yolov8的网络，可以便于后续改进如下图中所示。

以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。