Object detection is a fundamental yet challenging task in computer vision. Despite the great strides made over recent years, modern detectors may still produce unsatisfactory performance due to certain factors, such as non-universal object features and single regression manner. In this paper, we draw on the idea of mutual-assistance (MA) learning and accordingly propose a robust one-stage detector, referred as MADet, to address these weaknesses. First, the spirit of MA is manifested in the head design of the detector. Decoupled classification and regression features are reintegrated to provide shared offsets, avoiding inconsistency between feature-prediction pairs induced by zero or erroneous offsets. Second, the spirit of MA is captured in the optimization paradigm of the detector. Both anchor-based and anchor-free regression fashions are utilized jointly to boost the capability to retrieve objects with various characteristics, especially for large aspect ratios, occlusion from similar-sized objects, etc. Furthermore, we meticulously devise a quality assessment mechanism to facilitate adaptive sample selection and loss term reweighting. Extensive experiments on standard benchmarks verify the effectiveness of our approach. On MS-COCO, MADet achieves 42.5% AP with vanilla ResNet50 backbone, dramatically surpassing multiple strong baselines and setting a new state of the art.
Please refer to install.md for installation.
Please see get_started.md for the basic usage of MADet.
| Backbone | Lr schd | Box AP | Set | Config | Baidu Yun | Google Drive |
|---|---|---|---|---|---|---|
| R50-FPN | 1x | 42.5 | test | config | key: ks5i | model |
| R50-FPN | 1x | 42.2 | val | config | key: yatj | model |
The implementation of MADet is based on mmdetection.
This project is released under the Apache 2.0 license.
@ARTICLE{MADet,
author={Xie, Xingxing and Lang, Chunbo and Miao, Shicheng and Cheng, Gong and Li, Ke and Han, Junwei},
journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
title={Mutual-Assistance Learning for Object Detection},
year={2023},
volume={45},
number={12},
pages={15171-15184},
doi={10.1109/TPAMI.2023.3319634}
}