请参考 MMCV 的安装文档 来安装 NPU 版本的 MMCV。
以下展示单机四卡场景的运行指令:
bash tools/dist_train.sh configs/deeplabv3/deeplabv3_r50-d8_4xb2-40k_cityscapes-512x1024.py 4以下展示单机单卡下的运行指令:
python tools/train.py configs/deeplabv3/deeplabv3_r50-d8_4xb2-40k_cityscapes-512x1024.py| Model | mIoU | Config | Download |
|---|---|---|---|
| deeplabv3 | 78.85 | config | log |
| deeplabv3plus | 79.23 | config | log |
| hrnet | 78.1 | config | log |
| fcn | 74.15 | config | log |
| icnet | 69.25 | config | log |
| pspnet | 77.21 | config | log |
| unet | 68.86 | config | log |
| upernet | 77.81 | config | log |
| apcnet | 78.02 | config | log |
| bisenetv1 | 76.04 | config | log |
| bisenetv2 | 72.44 | config | log |
注意:
- 如果没有特别标记，NPU 上的使用混合精度训练的结果与使用 FP32 的 GPU 上的结果相同。
以上模型结果由华为昇腾团队提供