[发明专利]一种基于多尺度特征融合的目标检测方法

权利要求书

1.一种基于多尺度特征融合的目标检测方法，其特征在于，包括以下步骤：

S1：构建目标检测模型，并对目标检测模型进行训练；

S2：获取待检测目标的图像数据；

S3：将待检测目标的图像数据输入经过训练的目标检测模型中；

所述目标检测模型首先基于图像数据提取多尺度的特征图；然后通过逐层特征融合和反馈特征融合的方式对低层特征图进行充分融合，得到对应的低层融合特征图；再通过注意力机制仅对高层特征图进行融合更新，得到对应的高层融合特征图；最后基于低层融合特征图和高层融合特征图生成对应的多尺度融合特征图；

S4：基于多尺度融合特征图完成目标检测。

2.如权利要求1所述的基于多尺度特征融合的目标检测方法，其特征在于：所述目标检测模型包括FPN网络、FFE网络和H-BFP网络；

FPN网络，用于基于图像数据提取多尺度的特征图，然后从最高层次开始向下进行逐层特征融合，并输出未进行特征融合的高层特征图，以及对低层特征图进行特征融合得到的初步融合特征图；

FFE网络，用于基于图像数据提取多尺度的特征图，然后将对应尺度的特征图与FPN网络输出的初步融合特征图进行逐层特征融合和反馈特征融合，以生成对应的低层融合特征图；

H-BFP网络，用于对高层特征图和低层融合特征图进行相加取平均，并提取对应的特征信息；然后基于特征信息和注意力机制对高层特征图进行融合更新，得到对应的高层融合特征图。

3.如权利要求2所述的基于多尺度特征融合的目标检测方法，其特征在于：FFE网络中包括反馈特征输入模块；反馈特征输入模块获取FPN网络输出的初步融合特征图，并通过两条分支实现初步融合特征图的输入；

分支一，用于将初步融合特征图作为反馈信息输入到FFE网络中，使得初步融合特征图能够与对应尺度的特征图进行逐层特征融合，以输出对应的深度融合特征图；

分支二，用于将初步融合特征图输入至FFE网络的输出位置，使得初步融合特征能够与FFE网络输出的深度融合特征图进行反馈特征融合，以生成对应的低层融合特征图。

4.如权利要求3所述的基于多尺度特征融合的目标检测方法，其特征在于：初步融合特征图与对应尺度的特征图进行逐层特征融合时，通过1*1卷积层调整FFE网络的输出，并将对应的通道大小调整为256，以输出对应的深度融合特征图。

5.如权利要求2所述的基于多尺度特征融合的目标检测方法，其特征在于：FFE网络中包括全局模块；全局模块用于对输入的初步融合特征图进行全局特征提取，以获取初步融合特征图中更丰富的特征语义信息。

6.如权利要求2所述的基于多尺度特征融合的目标检测方法，其特征在于：H-BFP网络首先对FPN网络输出的高层特征图P_h进行下采样，得到对应的高层特征图P_h+1；然后将高层特征图P_h和P_h+1与FFE网络输出的低层融合特征图进行相加取平均，得到各层融合后的特征信息；再对特征信息进行no local操作，并通过特征信息采样的方式得到与高层特征图P_h和P_h+1对应尺度的高层融合特征图P′_h和P′_h+1；再对高层融合特征图P′_h和P′_h+1进行卷积和激活函数操作，分别计算得到对应的注意力特征图；最后基于高层融合特征图P′_h和P′_h+1对应的注意力特征图分别计算P_h与P′_h的加权和，以及P_h+1与P′_h+1加权和，进而更新高层融合特征图P′_h和P′_h+1。