From 2ac2dd92c787e411d79a61a32f4fdb52902fc578 Mon Sep 17 00:00:00 2001
From: pengmian <pengmian@baidu.com>
Date: Thu, 12 Sep 2019 12:01:22 +0800
Subject: [PATCH] add model

---
 docs/model_zoo.md                  | 1 +
 pretrained_model/download_model.py | 4 +++-
 turtorial/finetune_pspnet.md       | 6 +++++-
 3 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/docs/model_zoo.md b/docs/model_zoo.md
index 817acd5d..e82d6c6c 100644
--- a/docs/model_zoo.md
+++ b/docs/model_zoo.md
@@ -42,3 +42,4 @@ train数据集合为Cityscapes训练集合，测试为Cityscapes的验证集合
 | DeepLabv3+/Xception65/gn  | Cityscapes |[deeplabv3p_xception65_cityscapes.tgz](https://paddleseg.bj.bcebos.com/models/deeplabv3p_xception65_cityscapes.tgz) |16|false| 0.7804 |
 | DeepLabv3+/Xception65/bn | Cityscapes |[Xception65_deeplab_cityscapes.tgz](https://paddleseg.bj.bcebos.com/models/xception65_bn_cityscapes.tgz) | 16 | false | 0.7715 |
 | 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 |
diff --git a/pretrained_model/download_model.py b/pretrained_model/download_model.py
index b175e17b..a4cee72c 100644
--- a/pretrained_model/download_model.py
+++ b/pretrained_model/download_model.py
@@ -56,7 +56,9 @@ model_urls = {
     "unet_bn_coco":
     "https://paddleseg.bj.bcebos.com/models/unet_coco_v3.tgz",
     "icnet_bn_cityscapes":
-    "https://paddleseg.bj.bcebos.com/models/icnet_cityscapes.tar.gz"
+    "https://paddleseg.bj.bcebos.com/models/icnet_cityscapes.tar.gz",
+    "pspnet50_cityscapes":
+    "https://paddleseg.bj.bcebos.com/models/pspnet50_cityscapes.tgz"
 }
 
 if __name__ == "__main__":
diff --git a/turtorial/finetune_pspnet.md b/turtorial/finetune_pspnet.md
index c4072b0c..85279a2a 100644
--- a/turtorial/finetune_pspnet.md
+++ b/turtorial/finetune_pspnet.md
@@ -64,6 +64,9 @@ DATASET:
 MODEL:
     MODEL_NAME: "pspnet"
     DEFAULT_NORM_TYPE: "bn"
+    PSPNET:
+        DEPTH_MULTIPLIER: 1
+        LAYERS: 50
 
 # 其他配置
 TRAIN_CROP_SIZE: (512, 512)
@@ -114,4 +117,5 @@ python pdseg/eval.py --use_gpu --cfg ./configs/test_pet.yaml
 
 |预训练模型名称|BackBone|Norm|数据集|配置|
 |-|-|-|-|-|
-|pspnet50_cityscapes|-|bn|Cityscapes|MODEL.MODEL_NAME: pspnet <br> MODEL.DEFAULT_NORM_TYPE: bn|
+|pspnet50_cityscapes|-|bn|Cityscapes|MODEL.MODEL_NAME: pspnet <br> MODEL.DEFAULT_NORM_TYPE: bn <br> MODEL.PSPNET.LAYERS: 50|
+|pspnet101_cityscapes|-|bn|Cityscapes|MODEL.MODEL_NAME: pspnet <br> MODEL.DEFAULT_NORM_TYPE: bn <br> MODEL.PSPNET.LAYERS: 101|
-- 
GitLab