【yolov5】yolov5.yaml配置文件详解

yolov5.yaml 是 YOLOv5 模型的配置文件,其中包含了模型的结构、超参数、训练和推理相关的设置等信息。该文件可以用来训练和推理 YOLOv5 模型。
以下是yolov5.yaml的实例:

# YOLOv5  by Ultralytics, GPL-3.0 license

# Parameters
nc: 80  # number of classes
depth_multiple: 0.33  # model depth multiple
width_multiple: 0.50  # layer channel multiple
anchors:
  - [10,13, 16,30, 33,23]  # P3/8
  - [30,61, 62,45, 59,119]  # P4/16
  - [116,90, 156,198, 373,326]  # P5/32

# YOLOv5 v6.0 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, 9, C3, [512]],
   [-1, 1, Conv, [1024, 3, 2]],  # 7-P5/32
   [-1, 3, C3, [1024]],
   [-1, 1, SPPF, [1024, 5]],  # 9
  ]

# YOLOv5 v6.0 head
head:
  [[-1, 1, Conv, [512, 1, 1]],
   [-1, 1, nn.Upsample, [None, 2, 'nearest']],
   [[-1, 6], 1, Concat, [1]],  # cat backbone P4
   [-1, 3, C3, [512, False]],  # 13

   [-1, 1, Conv, [256, 1, 1]],
   [-1, 1, nn.Upsample, [None, 2, 'nearest']],
   [[-1, 4], 1, Concat, [1]],  # cat backbone P3
   [-1, 3, C3, [256, False]],  # 17 (P3/8-small)

   [-1, 1, Conv, [256, 3, 2]],
   [[-1, 14], 1, Concat, [1]],  # cat head P4
   [-1, 3, C3, [512, False]],  # 20 (P4/16-medium)

   [-1, 1, Conv, [512, 3, 2]],
   [[-1, 10], 1, Concat, [1]],  # cat head P5
   [-1, 3, C3, [1024, False]],  # 23 (P5/32-large)

   [[17, 20, 23], 1, Detect, [nc, anchors]],  # Detect(P3, P4, P5)
  ]

nc:需要检测的类别数。
depth_multiple:控制网络深度的因子,通常取值为 0.33、0.67、1.0、1.33 等。
width_multiple:控制网络宽度的因子,通常取值为 0.25、0.33、0.5、0.67、0.75、1.0 等。
anchors:定义预设的锚框,用于检测目标的大小和比例。anchors 中包含的是 k-means 算法聚类得到的 9 个默认锚框的大小和比例。
backbone:主干网络的配置,包括网络的类型、输出通道数、层数等。
head:检测头的配置,包括卷积层的大小、激活函数、BN 层等。
yolo:YOLO 检测器的配置,包括预测框的数量、锚框的尺寸和比例等。

在 yolov5s.yaml 中,还包含了许多其它的配置参数,例如优化器的超参数、学习率的策略、训练过程中的数据增强等。这些参数可以通过修改 yolov5s.yaml 来优化模型的性能,从而更好地适应不同的检测任务和数据集。

[-1, 9, C3, [512]]

上方语句是卷积层参数,其中:

-1 表示使用上一层的输出特征图作为输入。
9 表示该层的输出通道数。
C3 表示该层所在的模块名称为 C3,C3 是 YOLOv5 的主干网络骨干部分之一,具体的结构可以参考 YOLOv5 的论文。
[512] 表示该卷积层的卷积核大小为 3 × 3 3\times3 3×3,同时有 512 512 512 个卷积核。

因此,该卷积层的输出特征图大小和上一层的输入特征图大小相同,而输出特征图的通道数为 9 9 9。具体的特征图大小可以根据前一层的输出特征图大小和卷积核大小来计算。

这是 YOLOv5s 的整个 backbone,由 10 层网络组成。下面是各层的具体说明:

[[-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, 9, C3, [512]],
   [-1, 1, Conv, [1024, 3, 2]],  # 7-P5/32
   [-1, 3, C3, [1024]],
   [-1, 1, SPPF, [1024, 5]],  # 9
  ]
第一层是一个卷积层,输出通道数为 64,卷积核大小为 6x6,步长为 2,填充为 2,输出特征图大小为输入的一半。
第二层是一个卷积层,输出通道数为 128,卷积核大小为 3x3,步长为 2,输出特征图大小为输入的一半。
第三层是一个 C3 模块,包含 3 个卷积层,每个卷积层的输出通道数为 128,卷积核大小分别为 1x1、3x3、1x1,不改变特征图大小。
第四层是一个卷积层,输出通道数为 256,卷积核大小为 3x3,步长为 2,输出特征图大小为输入的一半。
第五层是一个 C3 模块,包含 6 个卷积层,每个卷积层的输出通道数为 256,卷积核大小分别为 1x1、3x3、1x1,不改变特征图大小。
第六层是一个卷积层,输出通道数为 512,卷积核大小为 3x3,步长为 2,输出特征图大小为输入的一半。
第七层是一个 C3 模块,包含 9 个卷积层,每个卷积层的输出通道数为 512,卷积核大小分别为 1x1、3x3、1x1,不改变特征图大小。
第八层是一个卷积层,输出通道数为 1024,卷积核大小为 3x3,步长为 2,输出特征图大小为输入的一半。
第九层是一个 C3 模块,包含 3 个卷积层,每个卷积层的输出通道数为 1024,卷积核大小分别为 1x1、3x3、1x1,不改变特征图大小。
第十层是一个 SPPF 层,具有金字塔式空间池化(Spatial Pyramid Pooling),输出通道数为 1024,使用大小为 5x5 的金字塔空间池化。

总的来说,YOLOv5s 的 backbone 采用了一系列的卷积层和 C3 模块,不断缩小特征图的尺寸和增加通道数,最后通过 SPPF 层得到一个 1x1x1024 的张量,再通过若干个全连接层和激活函数得到预测框的位置和类别。。

卷积核大小为什么分别是1x1、3x3、1x1?

这种卷积核大小的结构被称为卷积块(Convolutional Block),它由三个不同大小的卷积核组成,分别为1x1、3x3、1x1。这种结构可以有效地提高模型的非线性拟合能力和特征表达能力,具有一定的优势。

其中,第一个1x1卷积层主要用于降低输入特征图的通道数,以减少计算量和参数数量,同时可以通过学习权重进行特征压缩,提高模型的表达能力;第二个3x3卷积层用于对特征进行空间卷积,增强特征表达能力;第三个1x1卷积层则用于恢复特征的通道数,同时可以将不同通道的特征进行组合,提高模型的表达能力。

这种结构被广泛应用于深度神经网络中,如ResNet、DenseNet等。

使用大小为 5x5 的金字塔空间池化有什么用

金字塔空间池化(pyramid spatial pooling)是一种多尺度池化技术,它可以用来提取图像特征。使用大小为5x5的金字塔空间池化可以从多个尺度下获取图像的特征信息,并且可以减少特征的尺寸,从而减少模型的参数数量和计算量。

具体来说,使用大小为5x5的金字塔空间池化可以将输入图像分别进行多次池化,每次池化的窗口大小分别为5x5、4x4、3x3、2x2和1x1。这样可以得到多个不同尺度的池化特征图。这些特征图可以被组合成一个多通道的特征图,用于后续的分类或检测任务。

金字塔空间池化可以提高模型对于不同尺度目标的感知能力,从而提高模型的性能。此外,金字塔空间池化可以在不增加模型复杂度的情况下提高模型的准确率。

从C3模块输入和输出的图像大小区别

在 YOLOv5 中,C3 模块是一个由 3 个卷积层组成的卷积块,其中第 1 和第 2 个卷积层的步幅为 2,会将输入的特征图大小减半,而第 3 个卷积层的步幅为 1,不会改变输入的特征图大小。因此,C3 模块的输出特征图大小为输入特征图大小的一半。

具体来说,如果输入的特征图大小为 H × W H \times W H×W,那么经过 C3 模块之后的输出特征图大小为 ⌊ H 2 ⌋ × ⌊ W 2 ⌋ × C \lfloor \frac{H}{2} \rfloor \times \lfloor \frac{W}{2} \rfloor \times C 2H×2W×C,其中 C C C 是卷积核的数量,通常为 512。

例如,在 YOLOv5 中,默认的输入图像大小为 640x640,经过 C3 模块之后,输出特征图大小为 320x320x512。

需要注意的是,在 YOLOv5 中,还有其他模块也会对特征图大小进行改变,例如 SPP 模块和 PANet 模块。因此,在理解 YOLOv5 的特征图处理流程时,需要考虑多个模块的影响。

你可能感兴趣的:(YOLO)