diff --git a/pretrained_model/download_model.py b/pretrained_model/download_model.py index a4cee72cb41014268644fdc11bb4856a500eed90..7b44c8038a6c09add5c1a8be292975a21b38fbb0 100644 --- a/pretrained_model/download_model.py +++ b/pretrained_model/download_model.py @@ -57,7 +57,7 @@ model_urls = { "https://paddleseg.bj.bcebos.com/models/unet_coco_v3.tgz", "icnet_bn_cityscapes": "https://paddleseg.bj.bcebos.com/models/icnet_cityscapes.tar.gz", - "pspnet50_cityscapes": + "pspnet50_bn_cityscapes": "https://paddleseg.bj.bcebos.com/models/pspnet50_cityscapes.tgz" } diff --git a/turtorial/finetune_pspnet.md b/turtorial/finetune_pspnet.md index e15e23cb1f27fc0811c6e5bdaa839e66dde7bc8f..34d20dda85cf56399710d0f70d610fed9997a042 100644 --- a/turtorial/finetune_pspnet.md +++ b/turtorial/finetune_pspnet.md @@ -21,7 +21,7 @@ python dataset/download_pet.py 接着下载对应的预训练模型 ```shell -python pretrained_model/download_model.py pspnet50_cityscapes +python pretrained_model/download_model.py pspnet50_bn_cityscapes ``` ## 三. 准备配置 @@ -117,5 +117,5 @@ python pdseg/eval.py --use_gpu --cfg ./configs/test_pet.yaml |预训练模型名称|BackBone|Norm|数据集|配置| |-|-|-|-|-| -|pspnet50_cityscapes|ResNet50|bn|Cityscapes|MODEL.MODEL_NAME: pspnet
MODEL.DEFAULT_NORM_TYPE: bn
MODEL.PSPNET.LAYERS: 50| -|pspnet101_cityscapes|ResNet101|bn|Cityscapes|MODEL.MODEL_NAME: pspnet
MODEL.DEFAULT_NORM_TYPE: bn
MODEL.PSPNET.LAYERS: 101| +|pspnet50_bn_cityscapes|ResNet50|bn|Cityscapes|MODEL.MODEL_NAME: pspnet
MODEL.DEFAULT_NORM_TYPE: bn
MODEL.PSPNET.LAYERS: 50| +|pspnet101_bn_cityscapes|ResNet101|bn|Cityscapes|MODEL.MODEL_NAME: pspnet
MODEL.DEFAULT_NORM_TYPE: bn
MODEL.PSPNET.LAYERS: 101|