基于视觉自注意力模型的无人机遥感弱小目标检测方法技术

技术编号:41301384 阅读:17 留言:0更新日期:2024-05-13 14:48
本发明专利技术公开了一种基于视觉自注意力模型的无人机遥感弱小目标检测方法,包括:将待检测遥感图像输入改进的YOLOV5网络模型得到检测结果;改进的YOLOV5网络模型包括:改进的主干特征提取网络用于通过引入的视觉自注意力机制从待检测遥感图像中提取第一特征图、第二特征图、第三特征图和第四特征图;改进的第一目标检测网络用于根据第二、第三和第四特征图提取第五特征图、第六特征图、第七特征图;新增的第二目标检测网络用于根据第一、第二和第五特征图并利用视觉自注意力机制提取第八特征图、弱小目标检测结果;改进的第一目标检测网络还用于根据第六、第七和第八特征图提取小、中和大目标检测结果。本发明专利技术提高了小目标检测准确性。

【技术实现步骤摘要】

本专利技术属于图像处理,具体涉及一种基于视觉自注意力模型的无人机遥感弱小目标检测方法


技术介绍

1、无人机遥感图像是重要的图像信息来源,实用性极高,无人机目标检测在军事、民用和科研等领域具有重要意义。首先,它提高了军事侦察和监视的效率,使军队能够更及时、精准地发现和追踪目标,提高作战实力。其次,无人机目标检测可用于边境巡逻和安全监控,加强国土安全防线,对于防范非法越境等问题具有积极作用。在民用领域,无人机目标检测可用于灾害监测、搜索救援和环境监测,提高了应对自然灾害和人为灾害的能力。

2、目前,国内外对遥感图像中弱小目标的检测主要分为经典机器学习方法与深度学习方法。传统的“基于手动特征提取+分类器”的目标检测算法己无法满足复杂环境和多尺度下的检测精度要求。随着深度学习在解决图像处理上表现出的高效实用性,利用深度学习实现航拍图像深度变化特征的提取可有效避免传统方法的缺陷。基于回归计算的单阶段目标检测方法,如yolov5,具备较好的检测性能、高效的速度、灵活的模型选择以及简便的训练和部署过程,使其成为当前目标检测领域的一个重要算法。针对无人机航拍本文档来自技高网...

【技术保护点】

1.一种基于视觉自注意力模型的无人机遥感弱小目标检测方法,其特征在于,所述方法包括:

2.根据权利要求1所述的基于视觉自注意力模型的无人机遥感弱小目标检测方法,其特征在于,所述改进的主干特征提取网络包括:依次连接的CBS模块、CBS模块、第一自注意力模块、CBS模块、第二自注意力模块、CBS模块、第三自注意力模块、CBS模块、第四自注意力模块和SPPF模块;其中,所述第一自注意力模块的输出为所述第一特征图;所述第二自注意力模块的输出为所述第二特征图;所述第三自注意力模块的输出为所述第三特征图;所述SPPF模块的输出为所述第四特征图。

3.根据权利要求2所述的基于视...

【技术特征摘要】

1.一种基于视觉自注意力模型的无人机遥感弱小目标检测方法,其特征在于,所述方法包括:

2.根据权利要求1所述的基于视觉自注意力模型的无人机遥感弱小目标检测方法,其特征在于,所述改进的主干特征提取网络包括:依次连接的cbs模块、cbs模块、第一自注意力模块、cbs模块、第二自注意力模块、cbs模块、第三自注意力模块、cbs模块、第四自注意力模块和sppf模块;其中,所述第一自注意力模块的输出为所述第一特征图;所述第二自注意力模块的输出为所述第二特征图;所述第三自注意力模块的输出为所述第三特征图;所述sppf模块的输出为所述第四特征图。

3.根据权利要求2所述的基于视觉自注意力模型的无人机遥感弱小目标检测方法,其特征在于,所述第一自注意力模块和所述第四自注意力模块网络结构相同,所述第一自注意力模块和所述第四自注意力模块包括3个依次连接的自注意力单元;

4.根据权利要求3所述的基于视觉自注意力模型的无人机遥感弱小目标检测方法,其特征在于,所述第一自注意力模块包括:依次连接的cbs模块、自注意力单元、自注意力单元、自注意力单元、拼接层、cbs模块,所述第一自注意力模块中第一个cbs模块的输入端作为所述第一自注意力模块的数据输入端,所述第一自注意力模块中第二个cbs模块的输出端作为所述第一自注意力模块的数据输出端,且所述第一自注意力模块中第一个cbs模块的输入端还通过一cbs模块与所述第一自注意力模块中拼接层的输入端连接。

5.根据权利要求3所述的基于视觉自注意力模型的无人机遥感弱小目标检测方法,其特征在于,所述第二自注意力模块包括:依次连接的cbs模块、自注意力单元、自注意力单元、自注意力单元、自注意力单元、自注意力单元、自注意力单元、拼接层、cbs模块,所述第二自注意力模块中第一个cbs模块的输入端作为所述第一自注意力模块的数据输入端,所述第二自注意力模块中第二个cbs模块的输出端作为所述第一自注意力模块的数据输出端,且所述第二自注意力模块中第一个cbs模块的输入端还通过一cbs模块与所述第二自注意力模块中拼接层的输入端连接。

6.根据权利要求3所述的基于视觉自注意力模型...

【专利技术属性】
技术研发人员:吴金建张子铮李北海
申请(专利权)人:西安电子科技大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1