Thanks to CIRKD (Cross-Image Relational Knowledge Distillation for Semantic Segmentation) We propose a new kd seg structure base on their code
Ubuntu 18.04 LTS
Python 3.8 (Anaconda is recommended)
CUDA 11.1
PyTorch 1.8.0
NCCL for CUDA 11.1
Install python packages:
pip install timm==0.3.2
pip install mmcv-full==1.2.7
pip install opencv-python==4.5.1.48
Backbones pretrained on ImageNet:
| CNN | Transformer |
|---|---|
| resnet101-imagenet.pth | mit_b0.pth |
| resnet18-imagenet.pth | mit_b1.pth |
| mobilenetv2-imagenet.pth | mit_b4.pth |
Support datasets:
| Dataset | Train Size | Val Size | Test Size | Class |
|---|---|---|---|---|
| Cityscapes | 2975 | 500 | 1525 | 19 |
| Pascal VOC Aug | 10582 | 1449 | -- | 21 |
| CamVid | 367 | 101 | 233 | 11 |
| ADE20K | 20210 | 2000 | -- | 150 |
| COCO-Stuff-164K | 118287 | 5000 | -- | 182 |