MMSegmentation-Distiller是一款先进的知识蒸馏工具箱,建立在MMSegmentation框架之上,致力于提升模型性能与效率,适用于各类语义分割任务。
分段细分该项目基于mmsegmentation(v-0.11.0),所有用法与相同,包括培训、测试等。蒸馏器动物园的安装设置如下:
新建conda环境:
```
conda create -n distiller python=3.7
```
安装PyTorch 1.3+
安装mmdetection-distiller
```shell
git clone https://github.com/pppppM/mmsegmentation-distiller.git
cd mmsegmentation-distiller
pip install -r requirements/build.txt
pip install -v -e .
```
训练(单GPU):
```shell
python tools/train.py configs/distiller/cwd/cwd_psp_r101-d8_distill_psp_r18_d8_
```