
MMSegmentation-Distiller:这是一个基于MMSegmentation的 knowledge distillation 工具箱
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
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_
```
全部评论 (0)
还没有任何评论哟~


