CBNet: A Composite Backbone Network Architecture for Object Detection
About
Modern top-performing object detectors depend heavily on backbone networks, whose advances bring consistent performance gains through exploring more effective network structures. In this paper, we propose a novel and flexible backbone framework, namely CBNetV2, to construct high-performance detectors using existing open-sourced pre-trained backbones under the pre-training fine-tuning paradigm. In particular, CBNetV2 architecture groups multiple identical backbones, which are connected through composite connections. Specifically, it integrates the high- and low-level features of multiple backbone networks and gradually expands the receptive field to more efficiently perform object detection. We also propose a better training strategy with assistant supervision for CBNet-based detectors. Without additional pre-training of the composite backbone, CBNetV2 can be adapted to various backbones (CNN-based vs. Transformer-based) and head designs of most mainstream detectors (one-stage vs. two-stage, anchor-based vs. anchor-free-based). Experiments provide strong evidence that, compared with simply increasing the depth and width of the network, CBNetV2 introduces a more efficient, effective, and resource-friendly way to build high-performance backbone networks. Particularly, our Dual-Swin-L achieves 59.4% box AP and 51.6% mask AP on COCO test-dev under the single-model and single-scale testing protocol, which is significantly better than the state-of-the-art result (57.7% box AP and 50.2% mask AP) achieved by Swin-L, while the training schedule is reduced by 6$\times$. With multi-scale testing, we push the current best single model result to a new record of 60.1% box AP and 52.3% mask AP without using extra training data. Code is available at https://github.com/VDIGPKU/CBNetV2.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Object Detection | COCO 2017 (val) | AP46.7 | 2454 | |
| Object Detection | COCO (test-dev) | mAP60.1 | 1195 | |
| Object Detection | MS COCO (test-dev) | -- | 677 | |
| Object Detection | COCO (val) | mAP59.1 | 613 | |
| Object Detection | COCO v2017 (test-dev) | mAP59.6 | 499 | |
| Instance Segmentation | COCO (val) | APmk51.8 | 472 | |
| Instance Segmentation | COCO (test-dev) | -- | 380 | |
| Instance Segmentation | COCO 2017 (test-dev) | -- | 253 | |
| Object Detection | MS-COCO 2017 (val) | -- | 237 | |
| Object Detection | COCO mini (val) | AP59.6 | 123 |