CBNet: A Composite Backbone Network Architecture for Object Detection

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.

PDF Abstract

Results from the Paper


Ranked #6 on Object Detection on COCO-O (using extra training data)

     Get a GitHub badge
Task Dataset Model Metric Name Metric Value Global Rank Uses Extra
Training Data
Result Benchmark
Object Detection COCO minival CBNetV2 (Dual-Swin-L HTC, single-scale) box AP 59.1 # 30
Instance Segmentation COCO minival CBNetV2 (Dual-Swin-L HTC, multi-scale) mask AP 51.8 # 16
Object Detection COCO minival CBNetV2 (Dual-Swin-L HTC, multi-scale) box AP 59.6 # 26
Instance Segmentation COCO minival CBNetV2 (Dual-Swin-L HTC, single-scale) mask AP 51 # 18
Object Detection COCO-O CBNetV2 (Swin-L) Average mAP 39.0 # 6
Effective Robustness 12.36 # 7
Instance Segmentation COCO test-dev CBNetV2 (Dual-Swin-L HTC, single-scale) mask AP 51.6 # 16
Instance Segmentation COCO test-dev CBNetV2 (Dual-Swin-L HTC, multi-scale) mask AP 52.3 # 14
Object Detection COCO test-dev CBNetV2 (Dual-Swin-L HTC, multi-scale) box mAP 60.1 # 27
Object Detection COCO test-dev CBNetV2 (Dual-Swin-L HTC, single-scale) box mAP 59.4 # 28

Methods