Remote Sensing Image Object Detection Based on Bidirectional Feature Fusion and Feature Selection

被引:0
|
作者
Xiao J.-S. [1 ]
Zhang S.-H. [1 ]
Chen Y.-H. [2 ]
Wang Y.-F. [1 ]
Yang L.-H. [1 ]
机构
[1] School of Electronic Information, Wuhan University, Wuhan
[2] School of Computer Science and Technology, Guangdong University of Technology, Guangzhou
来源
关键词
Angle prediction; Feature fusion network; Multiple feature selection; Object detection; Remote sensing image;
D O I
10.12263/DZXB.20210354
中图分类号
学科分类号
摘要
In remote sensing image object detection, the complex background always occupies a large area of the entire image, which seriously affects the object detection effect. This paper proposes an object detection network that can perform multiple feature fusion and selection on feature maps. A feature fusion network is used to fuse deep and shallow features to improve the detection effect of small objects in complex background. While retaining the up-bottom path of the feature fusion network, it adds a bottom-up path to diminish the number of network layers that the shallow features need to pass on to the top layer, thereby reducing the loss of shallow features. In order to reduce the interference of useless information in the fusion feature maps with detection network, a multiple feature selection module is designed. The attention mechanism in the multiple feature selection module enables the network to adaptively focus on more important features, ignore useless features. Since the conventional five-parameter regression method has serious boundary problems, the angle prediction is often inaccurate for objects with a large aspect ratio, to solve this problem, the proposed method treats angle prediction as a classification task. The mAP of our method on DOTA and self-made dataset DOTA-GF reaches 0.651 and 0.641, and the comparative experiments with mainstream object detection methods demonstrate the effectiveness of the proposed method. © 2022, Chinese Institute of Electronics. All right reserved.
引用
收藏
页码:267 / 272
页数:5
相关论文
共 15 条
  • [1] PEI Wei, XU Yan-ming, ZHU Yong-ying, Et al., The target detection method of aerial photography images with improved SSD, Journal of Software, 30, 3, pp. 738-758, (2019)
  • [2] LIU Xiao-bo, LIU Peng, Et al., Research progress of optical remote sensing image object detection based on deep learning, Acta Automatica Sinica, 47, 9, pp. 2078-2089, (2021)
  • [3] XIAO Jin-sheng, ZHOU Jin-long, LEI Jun-feng, Et al., Single image dehazing algorithm based on the learning of hazy layers, Acta Electronica Sinica, 47, 10, pp. 2142-2148, (2019)
  • [4] XIAO Jin-sheng, ZHANG Shu-hao, DAI Yuan, Et al., A multiclass object detection in UAV images based on rotation region network, IEEE Journal on Miniaturization for Air and Space Systems, 1, 3, pp. 188-196, (2020)
  • [5] MA Jian-qi, SHAO Wei-yuan, YE Hao, Et al., Arbitrary-oriented scene text detection via rotation proposals, IEEE Transactions on Multimedia, 20, 11, pp. 3111-3122, (2018)
  • [6] LIN T Y, DOLLAR P, GIRSHICK R, Et al., Feature pyramid networks for object detection, 2017 IEEE Conference on Computer Vision and Pattern Recognition(CVPR), pp. 2117-2125, (2017)
  • [7] DAI Yuan, YI Ben-shun, XIAO Jin-sheng, Et al., Object detection of remote sensing image based on improved rotation region proposal network, Acta Optica Sinica, 40, 1, pp. 270-280, (2020)
  • [8] YANG Xue, YAN Chun-li, Feng Zi-ming, Et al., R3Det: refined single-stage detector with feature refinement for rotating object, 35th AAAI Conference on Artificial Intelligence, pp. 3163-3171, (2021)
  • [9] YANG Xue, YANG Ji-rui, YAN Jun-chi, Et al., SCRDet: Towards more robust detection for small, cluttered and rotated objects, 2019 IEEE/CVF International Conference on Computer Vision(ICCV), (2019)
  • [10] XIA Gui-song, BAI Xiang, DING Jian, Et al., DOTA: A large-scale dataset for object detection in aerial images, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 3974-3983, (2018)