Yolov5的yaml文件解读

Yolov5网络架构解析：参数配置与模块详解

最新推荐文章于 2025-04-24 16:03:22 发布

原创

最新推荐文章于 2025-04-24 16:03:22 发布 · 971 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#YOLO

本文为［365天深度学习训练营］(https://mp.weixin.qq.com/s/0dvHCaOoFnW8SCp3JpzKxg）中的学习记录客
原作者：[K同学啊］https://mtyjkh.blog.youkuaiyun.com/

最新Ultralytics给了Yolov5的两个网络模型，分别是yolov5和yolov5-p6两个部分，本文选择yolov5作为研究对象进行介绍。

yolov5.yaml文件是yolov5网络结构的定义文件，像改进算法的需要在该文件修该相关参数，让后再修改common.py和yolo.py的相关文件。

1.参数配置

这里面nc是分类的数目，scales是不同模型的信息，包括深度，宽度和最大通道数目

2.backbone

backbone:
  # [from, number, module, args]
  - [-1, 1, Conv, [64, 6, 2, 2]] # 0-P1/2
  - [-1, 1, Conv, [128, 3, 2]] # 1-P2/4
  - [-1, 3, C3, [128]]
  - [-1, 1, Conv, [256, 3, 2]] # 3-P3/8
  - [-1, 6, C3, [256]]
  - [-1, 1, Conv, [512, 3, 2]] # 5-P4/16
  - [-1,