From 3c65454c4c5606a4d845934f4df44f70b4144167 Mon Sep 17 00:00:00 2001 From: pennypm <34053581+pennypm@users.noreply.github.com> Date: Thu, 10 Oct 2019 15:59:53 +0800 Subject: [PATCH] add pspnet coco model (#53) * add pspnet coco model --- docs/model_zoo.md | 3 +++ pretrained_model/download_model.py | 8 +++++++- 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/docs/model_zoo.md b/docs/model_zoo.md index 612f74e6..01d535b4 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 7b44c803..6d7c265f 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__": -- GitLab