Home

Awesome

If you want to distill model in OpenMMLab related repos, please use MMRazor!!

If you are intrested in KD,you also could contact me by Wechat, and I will invite you to the KD group.

<img src="https://github.com/pppppM/mmsegmentation-distiller/blob/master/QR.jpeg" width="150" height="200" ><br/>

This project is based on mmdetection(v-2.9.0), all the usage is the same as mmdetection including training , test and so on.

Distiller Zoo

Installation

Train

#single GPU
python tools/train.py configs/distillers/cwd/cwd_retina_rx101_64x4d_distill_retina_r50_fpn_2x_coco.py

#multi GPU
bash tools/dist_train.sh configs/distillers/cwd/cwd_retina_rx101_64x4d_distill_retina_r50_fpn_2x_coco.py 8

Test

#single GPU
python tools/test.py configs/distillers/cwd/cwd_retina_rx101_64x4d_distill_retina_r50_fpn_2x_coco.py $CHECKPOINT --eval bbox

#multi GPU
bash tools/dist_train.sh configs/distillers/cwd/cwd_retina_rx101_64x4d_distill_retina_r50_fpn_2x_coco.py $CHECKPOINT 8 --eval bbox

Lisence

This project is released under the Apache 2.0 license.