diff --git a/docs/model_zoo.md b/docs/model_zoo.md index 612f74e6629870b96e7dbdf1f514958a34a4bd25..01d535b4b11a3e7e34f1939edb05fc9994307343 100644 --- a/docs/model_zoo.md +++ b/docs/model_zoo.md @@ -31,6 +31,8 @@ PaddleSeg对所有内置的分割模型都提供了公开数据集下的预训 | DeepLabv3+/MobileNetv2/bn | COCO |[deeplab_mobilenet_x1_0_coco.tgz](https://bj.bcebos.com/v1/paddleseg/deeplab_mobilenet_x1_0_coco.tgz) | 16 | --| -- | | DeeplabV3+/Xception65/bn | COCO | [xception65_coco.tgz](https://paddleseg.bj.bcebos.com/models/xception65_coco.tgz)| 16 | -- | -- | | U-Net/bn | COCO | [unet_coco.tgz](https://paddleseg.bj.bcebos.com/models/unet_coco_v3.tgz) | 16 | -- | -- | +| PSPNet/bn | COCO | [pspnet50_coco.tgz](https://paddleseg.bj.bcebos.com/models/pspnet50_coco.tgz) | 16 | -- | -- | +| PSPNet/bn | COCO | [pspnet101_coco.tgz](https://paddleseg.bj.bcebos.com/models/pspnet101_coco.tgz) | 16 | -- | -- | ## Cityscapes预训练模型 @@ -43,3 +45,4 @@ train数据集合为Cityscapes训练集合,测试为Cityscapes的验证集合 | DeepLabv3+/Xception65/bn | Cityscapes |[deeplabv3p_xception65_bn_cityscapes_.tgz](https://paddleseg.bj.bcebos.com/models/xception65_bn_cityscapes.tgz) | 16 | false | 0.7930 | | ICNet/bn | Cityscapes |[icnet_cityscapes.tgz](https://paddleseg.bj.bcebos.com/models/icnet6831.tar.gz) |16|false| 0.6831 | | PSPNet/bn | Cityscapes |[pspnet50_cityscapes.tgz](https://paddleseg.bj.bcebos.com/models/pspnet50_cityscapes.tgz) |16|false| 0.6968 | +| PSPNet/bn | Cityscapes |[pspnet101_cityscapes.tgz](https://paddleseg.bj.bcebos.com/models/pspnet101_cityscapes.tgz) |16|false| 0.7736 | diff --git a/pretrained_model/download_model.py b/pretrained_model/download_model.py index 7b44c8038a6c09add5c1a8be292975a21b38fbb0..6d7c265f6514ee3d16aa8e010ba9b071031ef07b 100644 --- a/pretrained_model/download_model.py +++ b/pretrained_model/download_model.py @@ -45,6 +45,10 @@ model_urls = { "https://paddleseg.bj.bcebos.com/models/xception65_coco.tgz", "unet_bn_coco": "https://paddleseg.bj.bcebos.com/models/unet_coco_v3.tgz", + "pspnet50_bn_coco": + "https://paddleseg.bj.bcebos.com/models/pspnet50_coco.tgz", + "pspnet101_bn_coco": + "https://paddleseg.bj.bcebos.com/models/pspnet101_coco.tgz", # Cityscapes pretrained "deeplabv3p_mobilenetv2-1-0_bn_cityscapes": @@ -58,7 +62,9 @@ model_urls = { "icnet_bn_cityscapes": "https://paddleseg.bj.bcebos.com/models/icnet_cityscapes.tar.gz", "pspnet50_bn_cityscapes": - "https://paddleseg.bj.bcebos.com/models/pspnet50_cityscapes.tgz" + "https://paddleseg.bj.bcebos.com/models/pspnet50_cityscapes.tgz", + "pspnet101_bn_cityscapes": + "https://paddleseg.bj.bcebos.com/models/pspnet101_cityscapes.tgz", } if __name__ == "__main__":