计算机工程与应用››2024,第60卷››发行(10): 237-245.内政部:10.3778/j.issn.1002-8331.2308-0196

图形图像处理 • 上一篇   下一篇

改进YOLOv7公司的城市小型无人机目标检测方法

崔勇强,李嘉轩,侯林果,梅涛,白迪,陈少平  

  1. 中南民族大学 电子信息工程学院,武汉 430074
  • 出版日期:2024-05-15 发布日期:2024-05-15

改进的YOLOv7小型城市无人机目标检测方法

崔永强、李家璇、侯林戈、梅涛、白迪、陈绍平  

  1. 中南民族大学电子与信息工程学院,武汉430074
  • 在线:2024-05-15 出版:2024-05-15

摘要:针对低小动”无人机的反制技术已成为低空空域安全防御的重要手段,然而实时检测与准确识别是实施有效反制的前提条件与关键基础。针对城市低空环境下,目标检测算法对不同背景下小尺度无人机目标检测精度低,容易出现漏检误检且易受外界因素干扰等问题,提出了一种基于改进YOLOv7“低小动”无人机目标检测方法。首先采集大量不同环境、不同背景下的无人机样本构建数据集,并采用ViBe(可视背景提取器)算法进行预处理;其次引入坐标注意力机制与SPDConv(空间到深度卷积)模块改进和优化YOLOv7公司的网络结构;最后提出融合ViBe和YOLOv7的二级检测架构,将改进后的YOLOv7公司作为网络模型检测经震波处理后的图像。依据原图与处理图像的位置大小关系,将检测出的目标坐标映射回归至原图片,从而完成目标检测提取。实验结果表明,所提目标检测方法检测精度达96.5%,约洛夫15.8个百分点,显著提升了低小动”目标的检测精度,能够满足低空无人机的实时精准检测的需求。

关键词: ViBe咖啡, 反无人机, YOLOv7, 坐标注意力机制, 小目标检测, SPDConv公司

摘要:针对“低空和小型移动”无人机的对抗已成为低空空域安全防御的重要工具,但实时检测和准确识别是有效对抗的前提和关键基础。针对城市低空环境,目标检测算法在检测不同背景下的小型无人机目标时精度较低,容易遗漏和误检,易受外界因素等干扰,“低空小移动”提出了基于改进YOLOv7的无人机目标检测方法。首先,从不同的环境和背景中采集大量的无人机样本,构建数据集,并使用ViBe(visual background extractor)算法进行预处理。其次,引入坐标注意机制和SPDConv(空间到深度卷积)模块来改进和优化YOLOv7的网络结构。最后,提出了一种二次检测架构,将ViBe和改进的YOLOv7融合,并使用改进的YLOv7作为网络模型检测ViBe处理的图像。基于原始图像和处理图像之间的位置-尺寸关系,将检测到的目标坐标映射回原始图像,从而完成目标检测和提取。实验结果表明,该目标检测方法的检测准确率达到96.5%,为15.8?比原来的YOLOv7方法提高了百分之几,显著提高了“低空和小型运动”目标的检测精度,满足了低空无人机实时精确检测的需求。

关键词: ViBe算法, 无人机, YOLOv7, 协调注意机制, 小目标检测, SPDConv公司