Our semantic segmentation implementation is based on MMSegmentation v0.19.0 and PVT segmentation. Thank the authors for their wonderful works.
Install MMSegmentation v0.19.0.
Prepare ADE20K according to the guidelines in MMSegmentation.
| Method | Backbone | Pretrain | Iters | mIoU | Download |
|---|---|---|---|---|---|
| Semantic FPN | EfficientMod-s-Conv | ImageNet-1K | 40K | 43.5 | [checkpoint & log] |
| Semantic FPN | EfficientMod-s | ImageNet-1K | 40K | 46.0 | [checkpoint & log] |
To evaluate EfficientMod + Semantic FPN on a single node with 8 GPUs run:
dist_test.sh configs/sem_fpn/{configure-file}.py /path/to/checkpoint_file 8 --out results.pkl --eval mIoU
To train EfficientMod + Semantic FPN on a single node with 8 GPUs run:
dist_train.sh configs/sem_fpn/{configure-file}.py 8