计算机工程与应用 ›› 2021, Vol. 57 ›› Issue (9): 212-216.DOI: 10.3778/j.issn.1002-8331.2002-0348

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

联合多尺度和注意力机制的遥感影像检测

张朕通,单玉刚,袁杰   

  1. 1.新疆大学 电气工程学院,乌鲁木齐 830047
    2.湖北文理学院 教育学院,湖北 襄阳 441053
  • 出版日期:2021-05-01 发布日期:2021-04-29

Remote Sensing Image Detection Algorithm Combining Multi-scale and Attention Mechanism

ZHANG Zhentong, SHAN Yugang, YUAN Jie   

  1. 1.College of Electrical Engineering, Xinjiang University, Urumqi 830047, China
    2.College of Education, Hubei University of Arts and Science, Xiangyang, Hubei 441053, China
  • Online:2021-05-01 Published:2021-04-29

摘要:

遥感影像中目标的检测问题一直是遥感图像处理领域的热点和难点。传统的检测算法,在解决场景复杂,尺度差异大的目标时性能不高,而使用深度学习很难兼顾遥感目标的准确性和实时性。针对这一问题,设计了一种利用多尺度融合特征检测目标的轻量级网络,并提出一种能够从三个维度上生成像素自适应特征权重的注意力机制帮助提取显著特征,同时采用了最新的优化算法改善模型的性能,在减少计算量的同时保证了检测精度。实验结果表明,该模型MAP@0.5可达0.945,F1可达0.841,检测速度满足实时性要求。

关键词: 模式识别, 遥感影像, 目标检测, 注意力机制, 多尺度, 优化算法

Abstract:

The problem of target detection in remote sensing images has always been a hot and difficult point in the field of remote sensing image processing. Traditional detection algorithms have low performance when solving complex scenes with large scale differences. However, it is difficult to balance the accuracy and real-time performance of remote sensing targets using deep learning. In response to this problem, this paper designs a lightweight network that uses multi-scale fusion feature detection targets, and proposes an attention mechanism that can generate pixel adaptive feature weights from three dimensions to help extract salient features. The latest optimization algorithm is used to improve the performance of the model, while reducing the amount of calculations and ensuring the detection accuracy. Experimental results show that the model MAP@0.5 can reach 0.945 and F1 can reach 0.841, and the detection speed can meet the real-time requirements.

Key words: pattern recognition, remote sensing image, target detection, attention mechanism, multi-scale, optimization algorithm