From 8fbea65691541b05e1fea1fcf4252b6334c005e6 Mon Sep 17 00:00:00 2001
From: LielinJiang <jianglielin@baidu.com>
Date: Fri, 13 Mar 2020 08:33:10 +0000
Subject: [PATCH] fix bugs from issue

---
 configs/deeplabv3p_mobilenetv2_cityscapes.yaml | 2 ++
 pdseg/models/model_builder.py                  | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/configs/deeplabv3p_mobilenetv2_cityscapes.yaml b/configs/deeplabv3p_mobilenetv2_cityscapes.yaml
index afb973fd..6b6835d1 100644
--- a/configs/deeplabv3p_mobilenetv2_cityscapes.yaml
+++ b/configs/deeplabv3p_mobilenetv2_cityscapes.yaml
@@ -30,6 +30,8 @@ MODEL:
         BACKBONE: "mobilenetv2"
         ASPP_WITH_SEP_CONV: True
         DECODER_USE_SEP_CONV: True
+        ENCODER_WITH_ASPP: False
+        ENABLE_DECODER: False
 TRAIN:
     PRETRAINED_MODEL_DIR: u"pretrained_model/deeplabv3p_mobilenetv2-1-0_bn_coco"
     MODEL_SAVE_DIR: "saved_model/deeplabv3p_mobilenetv2_cityscapes"
diff --git a/pdseg/models/model_builder.py b/pdseg/models/model_builder.py
index 3ff7e1ea..668d69e4 100644
--- a/pdseg/models/model_builder.py
+++ b/pdseg/models/model_builder.py
@@ -85,7 +85,7 @@ def seg_model(image, class_num):
         logits = fast_scnn.fast_scnn(image, class_num)
     else:
         raise Exception(
-            "unknow model name, only support unet, deeplabv3p, icnet, pspnet, hrnet"
+            "unknow model name, only support unet, deeplabv3p, icnet, pspnet, hrnet, fast_scnn"
         )
     return logits
 
-- 
GitLab