提交 8fbea656 编写于 作者: L LielinJiang

fix bugs from issue

上级 a271f8b8
...@@ -30,6 +30,8 @@ MODEL: ...@@ -30,6 +30,8 @@ MODEL:
BACKBONE: "mobilenetv2" BACKBONE: "mobilenetv2"
ASPP_WITH_SEP_CONV: True ASPP_WITH_SEP_CONV: True
DECODER_USE_SEP_CONV: True DECODER_USE_SEP_CONV: True
ENCODER_WITH_ASPP: False
ENABLE_DECODER: False
TRAIN: TRAIN:
PRETRAINED_MODEL_DIR: u"pretrained_model/deeplabv3p_mobilenetv2-1-0_bn_coco" PRETRAINED_MODEL_DIR: u"pretrained_model/deeplabv3p_mobilenetv2-1-0_bn_coco"
MODEL_SAVE_DIR: "saved_model/deeplabv3p_mobilenetv2_cityscapes" MODEL_SAVE_DIR: "saved_model/deeplabv3p_mobilenetv2_cityscapes"
......
...@@ -85,7 +85,7 @@ def seg_model(image, class_num): ...@@ -85,7 +85,7 @@ def seg_model(image, class_num):
logits = fast_scnn.fast_scnn(image, class_num) logits = fast_scnn.fast_scnn(image, class_num)
else: else:
raise Exception( 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 return logits
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册