[发明专利]一种基于特征增强的复杂场景下小目标检测方法

权利要求书

1.一种基于特征增强的复杂场景下小目标检测方法,包括以下步骤

步骤1、数据准备：数据集来源于航拍图像；

步骤2、数据增强：提出Cutout-DA数据增强方法，该方法首先在数据集中任意选取部分数据图像，然后通过对这些图像中的可见部分目标和全部目标，随机地按照目标大小比例的0.2、0.4、0.6、0.8进行部分位置的遮挡，生成新的遮挡数据扩充至VisDrone2021数据集中；

步骤3、设计多尺度融合的特征增强路径聚合网络MSFE-PANet；

步骤3.1：改进网络预测尺度

在YOLOv4中移除针对检测大目标预测头YOLO head3，但保留其所对应的13*13的特征图；同时，在预测网络中增加了由浅层高分辨率的特征图104*104生成的针对检测小尺度目标的预测头YOLO head0,生成新的网络预测尺度结构；

步骤3.2：特征层融合

在新的网络预测尺度结构上将每一层特征网络提取的特征图进行相应倍数上采样分别与第一层特征图相加融合得到新的特征图；

步骤3.3：引入注意力模块；

步骤4：设计预测框排斥损失函数RB_Loss；

步骤5：训练模型。

2.根据权利要求1所述的一种基于特征增强的复杂场景下小目标检测方法,其特征在于：所述步骤3.3具体包括

步骤3.3.1：添加CBAM注意力模块，如公式(1)所示；

通道注意力的计算公式为(2)：其中σ为Sigmoid激活函数，MLP权重W₀和W₁是共享的

空间注意力的计算公式为(3)：其中σ为Sigmoid激活函数，f^7*7表示7*7的滤波器；

步骤3.3.2：改进CBAM的通道注意力模块；

步骤3.3.3：引入SE-attention注意力模块；

步骤3.3.4：改进SPP模块；

步骤3.3.5：优化SE-attention注意力模块。

3.根据权利要求2所述的一种基于特征增强的复杂场景下小目标检测方法,其特征在于：所述步骤3.3.2，计算公式定义为(4)

4.根据权利要求3所述的一种基于特征增强的复杂场景下小目标检测方法,其特征在于：所述步骤3.3.3，给定一个输入X,通道数为C₁,经过F_tr的一系列卷积、池化操作得到通道数为C₂的特征U；F_sq为特征压缩操作，顺着空间维度进行特征压缩，并将每个二维的特征通道变成一个像素；接着进行F_ex激励操作，然后通过乘法加权到之前的特征上

计算公式(5)：

其中：U_C表示特征图中的第C个通道；Z_C为压缩操作的输出，计算公式(6):σ为Sigmoid激活函数；W₁,W₂均为全连接操作；δ为ReLU激活函数，计算公式(7):S_C为步长S中的第C个权重；

S＝F_ex(Z,W)＝σ(g(Z,W))＝σ(W₂δ(W₁Z)) (6)

F_scale＝(U_C,S_C)＝S_C·U_C (7)。

5.根据权利要求4所述的一种基于特征增强的复杂场景下小目标检测方法,其特征在于：所述步骤3.3.4具体为

将SPP中的1*1，5*5，9*9和13*13尺寸内核的池化层改为1*1卷积和3*3空洞卷积，改进的SPP模块不会改变特征图的尺寸，输出特征图尺寸计算公式为(8)