Anchor-Free Network for Multi-class Object Detection in Remote Sensing Images

被引:0
作者
Zhao, Guochuan [1 ,2 ]
Pang, Jie [1 ,2 ]
Zhang, Hua [1 ,2 ,3 ]
Zhou, Jian [1 ,2 ]
Li, Linjing [1 ,2 ]
机构
[1] Robot Technol Used Special Environm Key Lab Sichu, Mianyang 621000, Sichuan, Peoples R China
[2] Southwest Univ Sci & Technol, Sch Informa Engn, Mianyang 621000, Sichuan, Peoples R China
[3] Tsinghua Univ, Sichuan Energy Internet Res Inst, Chengdu 610213, Sichuan, Peoples R China
来源
PROCEEDINGS OF THE 39TH CHINESE CONTROL CONFERENCE | 2020年
关键词
anchor-free; object detection; sensing image;
D O I
暂无
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The anchor-based object detection algorithms need many hyper-parameters artificial such as the threshold of intersection over union and the size of anchors, which may limit the detection performance to some extent. In order to better solve the problem of multi-class object detection in remote sensing images, this paper proposed an anchor-free object detection network that does not require any hyper-parameters artificial. The network improves the detection effect of small objects by using the improved feature pyramid network to fuse multi-scale feature maps more effectively, and Focal loss and loss of intersection over union (loll loss) are used as loss functions to optimize the network. The experimental results on the DOTA dataset show that the mean average precision (mAP) of this network is 71.02%, which is at least 10.5% higher than other existing networks. The anchor-free network proposed in this paper achieves superior detection performance in remote sensing images.
引用
收藏
页码:7510 / 7515
页数:6
相关论文
共 16 条
[1]  
[Anonymous], 2016, P COMPUTER VISION EC, DOI DOI 10.1007/978-3-319-46448-0_2
[2]  
Dai JF, 2016, ADV NEUR IN, V29
[3]   Fast R-CNN [J].
Girshick, Ross .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1440-1448
[4]   Rich feature hierarchies for accurate object detection and semantic segmentation [J].
Girshick, Ross ;
Donahue, Jeff ;
Darrell, Trevor ;
Malik, Jitendra .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587
[5]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[6]  
He KM, 2014, LECT NOTES COMPUT SC, V8691, P346, DOI [arXiv:1406.4729, 10.1007/978-3-319-10578-9_23]
[7]   Focal Loss for Dense Object Detection [J].
Lin, Tsung-Yi ;
Goyal, Priya ;
Girshick, Ross ;
He, Kaiming ;
Dollar, Piotr .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2020, 42 (02) :318-327
[8]   Feature Pyramid Networks for Object Detection [J].
Lin, Tsung-Yi ;
Dollar, Piotr ;
Girshick, Ross ;
He, Kaiming ;
Hariharan, Bharath ;
Belongie, Serge .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :936-944
[9]   YOLO9000: Better, Faster, Stronger [J].
Redmon, Joseph ;
Farhadi, Ali .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :6517-6525
[10]   You Only Look Once: Unified, Real-Time Object Detection [J].
Redmon, Joseph ;
Divvala, Santosh ;
Girshick, Ross ;
Farhadi, Ali .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :779-788