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

说明书

本发明具体涉及一种基于多尺度特征融合的目标检测方法，包括：构建目标检测模型并进行训练；获取待检测目标的图像数据；将待检测目标的图像数据输入经过训练的目标检测模型中；首先基于图像数据提取多尺度的特征图；然后通过逐层特征融合和反馈特征融合的方式对低层特征图进行充分融合，得到对应的低层融合特征图；再通过注意力机制仅对高层特征图进行融合更新，得到对应的高层融合特征图；最后基于低层融合特征图和高层融合特征图生成对应的多尺度融合特征图；基于多尺度融合特征图完成目标检测。本发明中基于多尺度特征融合的目标检测方法能够充分融合尺度特征并保证小目标的检测效果，从而能够提升目标检测的准确性和效果。

技术领域

本发明涉及目标检测技术领域，具体涉及一种基于多尺度特征融合的目标检测方法。

背景技术

目标检测技术是计算机视觉领域的一大核心研究方向，旨在获取图像中感兴趣目标的所属分类和所在位置。目标检测技术不仅是目标跟踪、语义分割等诸多计算机视觉任务的研究基础，也被广泛应用于医学诊断、自动驾驶、智能视频监控、军事目标监测等多种民用和军用领域。随着应用场景的多元化和复杂化，一张待检图像中会包含多个不同尺度的目标，这使得现阶段的目标检测任务面临着尺度差异带来的严峻挑战。因此，多尺度目标检测也因此成为了目标检测领域的研究热点之一。

针对现有目标检测方法特征信息利用率不高的问题，公开号为CN113361528A的中国专利公开了《一种多尺度目标检测方法及系统》，其包括：构建包括依次连接的多个卷积模块和多个卷积支路的空洞金字塔网络模型；各卷积模块的输出分别连接一个卷积支路，卷积支路包括一个卷积操作和多个空洞卷积操作，卷积支路中的卷积操作和空洞卷积操作为并列关系；卷积支路的输出按照分辨率从低到高，依次将各第一特征图进行上采样操作后与同尺寸的相邻第一特征图进行元素级相加，获得多个融合特征图；根据目标检测数据集对空洞金字塔网络模型进行模型训练，获得目标检测模型。

上述现有方案中的目标检测方法通过不同卷积核的空洞卷积提取特征信息，使得卷积层具有不同尺寸的感受野，有助于单一特征层获取到更丰富的多尺度上下文特征信息，进而能够增强各个特征层对不同尺度目标的敏感性。实际目标检测时，是将图像数据输入训练好的目标检测模型中，由目标检测模型获得对应的多个融合特征图，进而基于多个融合特征图完成目标检测。因此，融合特征图在一定程度上决定着目标检测的效果。然而，现有方案在获得融合特征图时，多数技术手段停留在骨干网络的改进与微观层面卷积方式的改进，未在宏观上对特征金字塔的多尺度特征融合方式做出有效改进，未使用相关手段来细化和平衡多尺度语义特征，其虽然能提取到包含多尺度特征信息，但难以充分地融合多尺度特征信息，尤其对于小目标(小尺度特征)而言，其目标检测效果会受到限制，进而导致目标检测的准确性难以得到保证。因此，如何设计一种能够充分融合尺度特征并保证小目标检测效果的目标检测方法是亟需解决的技术问题。

发明内容

针对上述现有技术的不足，本发明所要解决的技术问题是：如何提供一种基于多尺度特征融合的目标检测方法，以能够充分融合尺度特征并保证小目标检测效果，从而提升目标检测的准确性和效果。

为了解决上述技术问题，本发明采用了如下的技术方案：

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

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

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

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

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

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