Abstract
Object detection usually includes two parts: objection classification and location. At present, the popular object detectors usually use two detection heads: one head is used to predict classification score, and the other one is used to predict the bounding box (bbox), respectively. In this paper, we first stack classification head after feature extract convolutional neural networks of bbox regression head. Then, we establish the classification networks by using a bounding box feature. The bounding box feature is very useful when the classification head uses soft Intersection over Union (IoU) labels. In experiment parts, only using PASCAL VOC 2007 datasets, soft Centerness labels, and soft IoU labels get 50.06 mAP and 52.08 mAP on VOC 2007 test. Compared with FCOS, they have 1.08% and 1.12% improvements. Using PASCAL VOC 2007 and 2012 datasets, our Union A*B head gets 78.71mAP after 12 epochs training with ResNet-101 as backbone and FPN as the neck. Extensive experimental results show that the proposed algorithm is superior to other detection methods.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.