计算机工程与应用 ›› 2022, Vol. 58 ›› Issue (13): 241-248.DOI: 10.3778/j.issn.1002-8331.2203-0065

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

改进YOLOv3的遥感影像小目标检测方法

牛浩青,欧鸥,饶姗姗,马万民   

  1. 成都理工大学 信息科学与技术学院(网络安全学院),成都 610051
  • 出版日期:2022-07-01 发布日期:2022-07-01

Small Object Detection Method Based on Improved YOLOv3 in Remote Sensing Image

NIU Haoqing, OU Ou, RAO Shanshan, MA Wanmin   

  1. College of Information Science & Technology(College of Internet Security), Chengdu University of Technology, Chengdu 610051, China
  • Online:2022-07-01 Published:2022-07-01

摘要: 针对当前目标检测任务中对小目标检测识别率低,漏检率高的问题,提出一种基于门控通道注意力机制(EGCA)和自适应上采样模块的改进YOLOv3算法。该算法采用DarkNet-53作为主干网络进行图片基础特征提取;引入自适应上采样模块对低分辨率卷积特征图进行扩张,有效增强了不同尺度卷积特征图的融合效果;在三个尺度通道输出预测结果之前分别加入EGCA注意力机制以提高网络对小目标的特征表达和检测能力。将改进的算法在公开数据集RSOD(remote sensing object detection)上进行实验,小目标检测精度提升了8.2个百分点,最为显著,平均精度AP值达到56.3%,较原算法提升了7.9个百分点。实验结果表明,改进的算法相比于传统YOLOv3算法和其他算法拥有更好的小目标检测能力。

关键词: 小目标检测, YOLOv3算法, 注意力机制, 上采样

Abstract: To solve the problems of low recognition rate and high miss rate of small object detection in the current target detection task, an improved YOLOv3 algorithm based on gated channel attention mechanism(EGCA) and adaptive upsampling module is proposed. Firstly, darknet-53 is used as the backbone network for image basic feature extraction. Secondly, the adaptive upsampling module is introduced to expand the low-resolution convolution feature map, which effectively enhances the fusion effect of different scale convolution feature map. Finally, EGCA attention mechanism is added before the three scale channels output the prediction results to improve the feature expression and detection ability of the network to small objects. The improved algorithm is tested on the public data set RSOD(remote sensing object detection), the accuracy of small object detection is improved by 8.2 percentage points, which is the most significant. The average accuracy AP value reaches 56.3%, which is 7.9 percentage points higher than the original algorithm. Experimental results show that the improved algorithm has better small object detection ability than the traditional YOLOv3 algorithm and other algorithms.

Key words: small object detection, YOLOv3 algorithm, channel attention mechanism, upsampling