Instance-based dynamic label assignment for object detection

被引:1
作者
Peng, Zilu [1 ]
Shao, Mingwen [1 ]
Sun, Yuantao [1 ]
Liu, Zeting [1 ]
Li, Cunhe [1 ]
机构
[1] China Univ Petr, Coll Comp Sci & Technol, Qingdao, Peoples R China
基金
中国国家自然科学基金;
关键词
object detection; label assignment; feature extraction; deep convolutional neural networks;
D O I
10.1117/1.JEI.31.4.043009
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Existing label assignment strategies have achieved promising performance for providing learning targets, typically rely on statistical information and location characteristics. However, classical label assignment strategies usually lack a comprehensive quality evaluation criterion. As a result, the quality of positive samples is not reliable, which weakens the performance of detectors. We propose a dynamic label assignment strategy to provide higher quality positive samples for detection models. Specifically, we propose a quality assessment criteria of candidate samples, which uses a joint representation guided by intersection over union (IoU). Consequently, the supervised information of both branches is included using only one metric. Beside, we propose a partitioning approach to eliminate local redundant sampling, allowing the selected positive sample points to focus more on the overall information of the target. Tests on the COCO dataset show that our work improves the baseline by 2.2% AP without additional modeling and supervisory information. In addition, extensive experiments on the MS COCO test-dev dataset using different backbones demonstrate that our best model outperforms most of the existing representative methods.
引用
收藏
页数:14
相关论文
共 41 条
[1]   Cascade R-CNN: Delving into High Quality Object Detection [J].
Cai, Zhaowei ;
Vasconcelos, Nuno .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :6154-6162
[2]  
Chen S., 2022, J ELECTRON IMAGING, V21, DOI [10.1117/1.JEI.21.1.013009, DOI 10.1117/1.JEI.21.1.013009]
[3]   Global contextual attention for pure regression object detection [J].
Fan, Bingbing ;
Shao, Mingwen ;
Li, Yunhao ;
Li, Cunhe .
INTERNATIONAL JOURNAL OF MACHINE LEARNING AND CYBERNETICS, 2022, 13 (08) :2189-2197
[4]   TOOD: Task-aligned One-stage Object Detection [J].
Feng, Chengjian ;
Zhong, Yujie ;
Gao, Yu ;
Scott, Matthew R. ;
Huang, Weilin .
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, :3490-3499
[5]   Mutual Supervision for Dense Object Detection [J].
Gao, Ziteng ;
Wang, Limin ;
Wu, Gangshan .
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, :3621-3630
[6]   Fast R-CNN [J].
Girshick, Ross .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1440-1448
[7]   AugFPN: Improving Multi-scale Feature Learning for Object Detection [J].
Guo, Chaoxu ;
Fan, Bin ;
Zhang, Qian ;
Xiang, Shiming ;
Pan, Chunhong .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2020), 2020, :12592-12601
[8]  
He KM, 2020, IEEE T PATTERN ANAL, V42, P386, DOI [10.1109/ICCV.2017.322, 10.1109/TPAMI.2018.2844175]
[9]   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
[10]   Probabilistic Anchor Assignment with IoU Prediction for Object Detection [J].
Kim, Kang ;
Lee, Hee Seok .
COMPUTER VISION - ECCV 2020, PT XXV, 2020, 12370 :355-371