Optimization for Anchor-Free Object Detection via Scale-Independent GIoU Loss

被引:0
|
作者
Cui, Min [1 ]
Duan, Yiming [2 ]
Pan, Chun [1 ]
Wang, Jiaolong [1 ]
Liu, Haitao [3 ]
机构
[1] Jiangnan Univ, Sch Internet Things, Wuxi 200240, Peoples R China
[2] Beijing Inst Technol, Sch Automat, Beijing 100081, Peoples R China
[3] Shanghai Jiao Tong Univ, Sch Mech Engn, Shanghai 200240, Peoples R China
基金
中国国家自然科学基金;
关键词
Shape; Detectors; Object detection; Loss measurement; Convergence; Shape measurement; Optimization; Anchor-free; arbitrary-oriented object detection; regression loss function; scale independence;
D O I
暂无
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
Arbitrary-oriented target detection is widely used in optical remote-sensing image processing, and there have been lots of anchor-based detectors using horizontal bounding boxes. However, the image targets of various scales and shapes make it difficult to tune optimal anchor parameters, whereas the complex background and nonmaximum suppression (NMS) require well-aligned bounding box to predict dense targets. In this letter, a scale-independent IoU (SIoU) loss is proposed for bounding box regression, which can adaptively adjust the shape of predicted boxes and speed up the convergence. Besides, the regression branch of the fully convolutional one-stage object detector (FCOS) is refined to implement the novel intersection over union (IoU) loss for rotated bounding box regression. Extensive experiments on HRSC2016 and a large-scale dataset for object detection in aerial images (DOTA) show that our method obtains 88.1% mean average precision (mAP) under an IoU threshold of 0.5 on HRSC2016, which is 1.1% higher than generalized IoU (GIoU) loss and 0.7% than complete IoU (CIoU) loss.
引用
收藏
页数:5
相关论文
共 50 条
  • [41] FII-CenterNet: An Anchor-Free Detector With Foreground Attention for Traffic Object Detection
    Fan, Siqi
    Zhu, Fenghua
    Chen, Shichao
    Zhang, Hui
    Tian, Bin
    Lv, Yisheng
    Wang, Fei-Yue
    IEEE TRANSACTIONS ON VEHICULAR TECHNOLOGY, 2021, 70 (01) : 121 - 132
  • [42] Using Anchor-Free Object Detectors to Detect Surface Defects
    Liu, Jiaxue
    Zhang, Chao
    Li, Jianjun
    PROCESSES, 2024, 12 (12)
  • [43] Class-Aware Prediction: A Solution to Center Point Collision for Anchor-Free Object Detection in Aerial Images
    Yong, Yixing
    Wang, Jian
    Li, Fan
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21 : 1 - 5
  • [44] SRAF-Net: A Scene-Relevant Anchor-Free Object Detection Network in Remote Sensing Images
    Liu, Junmin
    Li, Shijie
    Zhou, Changsheng
    Cao, Xiangyong
    Gao, Yong
    Wang, Bo
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [45] SID: Incremental learning for anchor-free object detection via Selective and Inter-related Distillation
    Peng, Can
    Zhao, Kun
    Maksoud, Sam
    Li, Meng
    Lovell, Brian C.
    COMPUTER VISION AND IMAGE UNDERSTANDING, 2021, 210
  • [46] A Light Anchor-Free Detection Network for Remote Sensing Images via Heatmap-Saliency Distillation
    Zhang, Ruiyan
    Jiang, Xiujie
    An, Junshe
    Cui, Tianshu
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
  • [47] High-resolution network Anchor-free object detection method based on iterative aggregation
    Wang X.
    Li Z.
    Zhang H.
    Beijing Hangkong Hangtian Daxue Xuebao/Journal of Beijing University of Aeronautics and Astronautics, 2021, 47 (12): : 2533 - 2541
  • [48] Predictive Distillation Method of Anchor-Free Object Detection Model for Continual Learning
    Gang, Sumyung
    Chung, Daewon
    Lee, Joonjae
    APPLIED SCIENCES-BASEL, 2022, 12 (13):
  • [49] Anchor-free object detection network based on non-local operation
    Wu, Peng
    Wang, Chen
    Wang, Junxiao
    Wang, Yuanzhi
    MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 83 (19) : 56249 - 56259
  • [50] DSLA: Dynamic smooth label assignment for efficient anchor-free object detection
    Su, Hu
    He, Yonghao
    Jiang, Rui
    Zhang, Jiabin
    Zou, Wei
    Fan, Bin
    PATTERN RECOGNITION, 2022, 131