Oriented Object Detection in Aerial Images Based on the Scaled Smooth L1 Loss Function

被引:6
|
作者
Wei, Linhai [1 ]
Zheng, Chen [2 ]
Hu, Yijun [1 ]
机构
[1] Wuhan Univ, Sch Math & Stat, Wuhan 430072, Peoples R China
[2] Henan Univ, Sch Math & Stat, Kaifeng 475001, Peoples R China
基金
中国国家自然科学基金;
关键词
object detection; convolution network; loss function; remote sensing image; aerial image; DETECTION FRAMEWORK; VEHICLE DETECTION;
D O I
10.3390/rs15051350
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
Although many state-of-the-art object detectors have been developed, detecting small and densely packed objects with complicated orientations in remote sensing aerial images remains challenging. For object detection in remote sensing aerial images, different scales, sizes, appearances, and orientations of objects from different categories could most likely enlarge the variance in the detection error. Undoubtedly, the variance in the detection error should have a non-negligible impact on the detection performance. Motivated by the above consideration, in this paper, we tackled this issue, so that we could improve the detection performance and reduce the impact of this variance on the detection performance as much as possible. By proposing a scaled smooth L1 loss function, we developed a new two-stage object detector for remote sensing aerial images, named Faster R-CNN-NeXt with RoI-Transformer. The proposed scaled smooth L1 loss function is used for bounding box regression and makes regression invariant to scale. This property ensures that the bounding box regression is more reliable in detecting small and densely packed objects with complicated orientations and backgrounds, leading to improved detection performance. To learn rotated bounding boxes and produce more accurate object locations, a RoI-Transformer module is employed. This is necessary because horizontal bounding boxes are inadequate for aerial image detection. The ResNeXt backbone is also adopted for the proposed object detector. Experimental results on two popular datasets, DOTA and HRSC2016, show that the variance in the detection error significantly affects detection performance. The proposed object detector is effective and robust, with the optimal scale factor for the scaled smooth L1 loss function being around 2.0. Compared to other promising two-stage oriented methods, our method achieves a mAP of 70.82 on DOTA, with an improvement of at least 1.26 and up to 16.49. On HRSC2016, our method achieves an mAP of 87.1, with an improvement of at least 0.9 and up to 1.4.
引用
收藏
页数:23
相关论文
共 50 条
  • [21] Convolutional Neural Network Based Automatic Object Detection on Aerial Images
    Sevo, Igor
    Avramovic, Aleksej
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2016, 13 (05) : 740 - 744
  • [22] Lightweight Object Detection Networks for UAV Aerial Images Based on YOLO
    Li, Yanshan
    Wang, Jiarong
    Zhang, Kunhua
    Yi, Jiawei
    Wei, Miaomiao
    Zheng, Lirong
    Xie, Weixin
    CHINESE JOURNAL OF ELECTRONICS, 2024, 33 (04) : 997 - 1009
  • [23] Cross-Grid Label Assignment for Arbitrary-Oriented Object Detection in Aerial Images
    Rao, Xiaohan
    Zhou, Liming
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21 : 1 - 5
  • [24] Attention Mechanism-Based Object Detection Algorithm in Aerial Images
    Bai, Zongbao
    Zhang, Junju
    Gao, Yuan
    Hu, Youcheng
    LASER & OPTOELECTRONICS PROGRESS, 2023, 60 (12)
  • [25] Attention-based object detection with saliency loss in remote sensing images
    Wu, Qin
    Yuan, Xingxing
    Yao, Zikang
    Chai, Zhilei
    JOURNAL OF ELECTRONIC IMAGING, 2021, 30 (01)
  • [26] Task-Wise Sampling Convolutions for Arbitrary-Oriented Object Detection in Aerial Images
    Huang, Zhanchao
    Li, Wei
    Xia, Xiang-Gen
    Wang, Hao
    Tao, Ran
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, : 1 - 15
  • [27] Adaptive scale matching for remote sensing object detection based on aerial images
    Han, Lu
    Li, Nan
    Zhong, Zeyuan
    Niu, Dong
    Gao, Bingbing
    IMAGE AND VISION COMPUTING, 2025, 157
  • [28] Small object detection in UAV aerial images based on inverted residual attention
    Liu S.
    Liu Y.
    Sun Y.
    Li Y.
    Wang J.
    Beijing Hangkong Hangtian Daxue Xuebao/Journal of Beijing University of Aeronautics and Astronautics, 2023, 49 (03): : 514 - 524
  • [29] Orientation Robust Object Detection in Aerial Images Based on R-NMS
    Liu, Qing Qing
    Li, Jian Bin
    PROCEEDINGS OF THE 9TH INTERNATIONAL CONFERENCE OF INFORMATION AND COMMUNICATION TECHNOLOGY [ICICT-2019], 2019, 154 : 650 - 656
  • [30] Object Detection Based on Global-Local Saliency Constraint in Aerial Images
    Li, Chengyuan
    Luo, Bin
    Hong, Hailong
    Su, Xin
    Wang, Yajun
    Liu, Jun
    Wang, Chenjie
    Zhang, Jing
    Wei, Linhai
    REMOTE SENSING, 2020, 12 (09)