This repository is the official implementation of ICCV2023: Bridging Cross-task Protocol Inconsistency for Distillation in Dense Object Detection.
This repo is build on mmdetection 2.28.2
Please refer this link to build the environment (mmcv...).
Then execute the following command to install.
git clone https://github.com/TinyTigerPan/BCKD.git
cd BCKD
pip install -v -e .
For single GPU
python tools/train.py configs/bckd/bckd_r50_gflv1_r101_fpn_coco_1x.py
For multi GPU
bash tools/dist_train.sh configs/bckd/bckd_r50_gflv1_r101_fpn_coco_1x.py 8
For single GPU
python tools/test.py config_file ckpt_file --eval bbox
For multi GPU
bash tools/dist_test.sh configs_file ckpt_file 8
Teacher | Student | Schedule | mAP | download |
---|---|---|---|---|
GFocal-R50 | 1x | 40.1 | ||
GFocal-R101 | GFocal-R50 | 1x | 43.2 | log | ckpt |
GFocal-R34 | 1x | 38.9 | ||
GFocal-R101 | GFocal-R34 | 1x | 42.0 | log | ckpt |
GFocal-R18 | 1x | 35.8 | ||
GFocal-R101 | GFocal-R18 | 1x | 38.6 | log | ckpt |
@InProceedings{Yang_2023_ICCV,
author = {Yang, Longrong and Zhou, Xianpan and Li, Xuewei and Qiao, Liang and Li, Zheyang and Yang, Ziwei and Wang, Gaoang and Li, Xi},
title = {Bridging Cross-task Protocol Inconsistency for Distillation in Dense Object Detection},
booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
month = {October},
year = {2023},
pages = {17175-17184}
}