YOLOv5配置文件之 - yaml

在YOLOv5的目录中,models文件夹里存储了YOLO的模型配置。
./models/yolov5.yaml 定义了YOLOv5s网络结构的定义文件

yaml的主要内容

参数配置

  • nc: 80
    • 类别数量
  • depth_multiple: 0.33
    • 模型深度缩放因子
  • width_multiple: 0.50
    • 控制卷积特征图的通道个数

anchors配置

有三组数组,共9个尺度,分别对应了小目标,中等目标和大目标

另外YOLOv5在训练前使用K- means和遗传进化算法,找到与目前数据集的分布最吻合的anchors

backbone

定义了骨干网络的结构,其中第一个模块一行,每行四个参数构成。

  • from
    • 当前模块的输入来源,-1表示上一层
  • number
    • 模块的重复次数,是基准值,不同大小的模型(s,m,n,x,l)可能会对它应用不同的绽放因子
  • module
    • 模块类名,通过这个名字在common.py中寻找相应的类
  • args
    • 保存了需要传递给模块的参数

head

基本的逻辑和上面的backbone一样

问题

如果我们要将yolov5s网络模型中第4层C3*2修改为C3*1,第6层的C3*3修改为C3*2

解答

第4层的number原本是6,经过depth_multiple缩放后,得到结果为2,想把它的循环次数修改为1,就需要将number修改为3
第6层的number原本是9,经过depth_multiple缩放后,得到结果为3,想把它的循环次数修改为2,就需要将number修改为2
YOLOv5配置文件之 - yaml_第1张图片

你可能感兴趣的:(YOLO,YOLO,cnn,神经网络,深度学习)