Remote Sensing Image Target Detection Model Based on Attention and Feature Fusion

被引:0
|
作者
Wang Yani [1 ]
Wang Xili [1 ]
机构
[1] Shaanxi Normal Univ, Sch Comp Sci, Xian 710119, Shaanxi, Peoples R China
关键词
image processing; remote sensing images; attention branch; feature fusion; target detection; OBJECT DETECTION;
D O I
10.3788/L0P202158.0228003
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Aiming at the problem that remote sensing images with complex environmental backgrounds and small targets are difficult to perform accurate target detection, based on the single-stage detection model (SSD), a singlestage target detection model based on attention and feature fusion is proposed in this paper, which is mainly composed of detection branch and attention branch. First, the attention branch is added to the detection branch SSD. The fully convolutional network (FCN) of the attention branch obtains the location characteristics of the target to be detected through pixel-by-pixel regression. Second, by using the method of adding corresponding elements to the detection branch and attention branch, the feature fusion of detection branch and attention branch are carried out to obtain high-quality feature image with more detailed information and semantic information. Finally, soft nonmaximum suppression (Soft-NMS) is used as a post-processing part to further improve the accuracy of target detection. Experimental results show that the mean average accuracy of the model on the UCAS-AOD and NWPU VHR-10 data sets are 92.52% and 82.49%, respectively. Compared with other models, the detection efficiency of the model is higher.
引用
收藏
页数:9
相关论文
共 22 条
  • [1] Soft-NMS - Improving Object Detection With One Line of Code
    Bodla, Navaneeth
    Singh, Bharat
    Chellappa, Rama
    Davis, Larry S.
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, : 5562 - 5570
  • [2] Real-Time Detection Based on Improved Single Shot MultiBox Detector
    Chen Lili
    Zhang Zhengdao
    Peng Li
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2019, 56 (01)
  • [3] Learning Rotation-Invariant Convolutional Neural Networks for Object Detection in VHR Optical Remote Sensing Images
    Cheng, Gong
    Zhou, Peicheng
    Han, Junwei
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (12): : 7405 - 7415
  • [4] Multi-class geospatial object detection and geographic image classification based on collection of part detectors
    Cheng, Gong
    Han, Junwei
    Zhou, Peicheng
    Guo, Lei
    [J]. ISPRS JOURNAL OF PHOTOGRAMMETRY AND REMOTE SENSING, 2014, 98 : 119 - 132
  • [5] Fu CY, 2020, DSSD DECONVOLUTIONAL
  • [6] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [7] He KM, 2014, LECT NOTES COMPUT SC, V8691, P346, DOI [arXiv:1406.4729, 10.1007/978-3-319-10578-9_23]
  • [8] Jaderberg M, 2020, SPATIAL TRANSFORMER
  • [9] Object Detection Algorithm Guided by Dual Attention Models
    Ji Zhong
    Kong Qiankun
    Wang Jian
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2020, 57 (06)
  • [10] [李红艳 Li Hongyan], 2019, [中国图象图形学报, Journal of Image and Graphics], V24, P1400