YOLOV7学习记录-

时间:2024-03-12 19:39:09

YOLOV7的改进

提升推理速度

RepConv-BN模块 (创新点1)

Conv层和BN层融合

将1×1卷积核提升到3×3

通常我们的1*1卷积核可以用来升维和降维。当我们认为网络中特征训练不好的时候通常增加网络的分支进行升维和降维。

具体操作:
在这里插入图片描述

为什么要将11卷积核提升到33。因为英伟达的硬件对3*3的卷积计算支持的比较好,计算的更快。

英伟达的硬件对33的卷积计算支持的比较好,计算的更快的原因有以下几点:33的卷积核可以利用英伟达的 Tensor Core1,这是一种专门用于加速矩阵乘法和累加的硬件单元,可以大幅提高卷积运算的效率和吞吐量2。
33的卷积核可以使用英伟达的 cuDNN3 库,这是一种针对深度神经网络的高性能计算库,它提供了一些优化的算法和实现,例如 Winograd4 算法,可以减少卷积运算的乘法次数和内存访问5。
3
3的卷积核可以适应英伟达的 GPU架构,这是一种基于多个 流处理器 并行执行的硬件设计,它可以充分利用GPU的计算资源和带宽,避免出现空闲或者等待的情况。
综上所述,英伟达的硬件对33的卷积计算支持的比较好,计算的更快,这是由于它的硬件特性和软件优化相互配合,使得33的卷积核能够发挥出最大的性能。希望这能回答你的问题。

那为什么11的卷积核计算不如33的卷积核计算的快呢?

11的卷积核只能在通道维度上进行滤波操作,而不会改变特征图的高度和宽度,因此它不能有效地利用空间局部性,导致内存访问的开销较大12。
1
1的卷积核的参数量和计算量相对较小,因此它不能充分发挥英伟达的Tensor Core3的优势,这是一种专门用于加速矩阵乘法和累加的硬件单元,可以大幅提高卷积运算的效率和吞吐量4。
1*1的卷积核的计算粒度较细,因此它不能有效地利用英伟达的GPU架构5,这是一种基于多个流处理器并行执行的硬件设计,它可以充分利用GPU的计算资源和带宽,避免出现空闲或者等待的情况。
综上所述,11的卷积核计算不如33的卷积核计算计算的快,这是由于它的计算特性和英伟达的硬件特性不匹配,导致性能下降。

同时,将11的卷积核提升到33后,可以将网络的分支和主线部分(本身3*3)合并计算。

resnet残差链接转换

在这里插入图片描述

要完成的任务

在这里插入图片描述

正样本分配策略(创新点2)

通俗的讲就是在训练过程中将候选框的中心点往上下左右中的两个方向移动,增加正样本候选框的数量,并计算更match的预测框。
我们训练的时候正样本是比较少的,负样本是无数的。因此需要一些策略提高正样本数量。
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

yolov7有4个输出层,每层还是有3个anchor。辅助头只是训练中使用,与主头的数量和作用是一致的。唯一的区别是主头的偏移量是0.5.辅助头的偏移量是1.
在这里插入图片描述
在这里插入图片描述
用maxpooling做下采样比较好? 还是用conv stride=2做下采样比较好? YOLOV7中使用的策略如下,就是都做。
在这里插入图片描述

源码理解

EMA训练技巧

什么是EMA?

EMA训练技巧是指在深度学习中使用指数移动平均(Exponential Moving Average)对模型的参数做平均,以提高测试指标并增加模型鲁棒性。EMA的本质是对历史数据进行加权平均,其中每个数据点的权重随着它距离当前时间点的远近而不断减小。这样做的好处是可以有效地平滑时间序列数据,使其更加连续和稳定。