diff --git a/configs/deeplabv3p_mobilenetv2_cityscapes.yaml b/configs/deeplabv3p_mobilenetv2_cityscapes.yaml index afb973fda3d5d413fb342d0315d0d062de9d525f..6b6835d11eed2bf87d332a4ada44785f78a959ec 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 3ff7e1eacad3f649820d11b16f793d80e94b806b..668d69e44aeb91cc7705a79f092730ae6a1fdb09 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