未验证 提交 3ce0777a 编写于 作者: L LielinJiang 提交者: GitHub

Merge pull request #193 from LielinJiang/fix-bugs-0313

Fix some bugs
...@@ -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"
......
...@@ -99,4 +99,4 @@ cd /d D:\projects\PaddleSeg\deploy\cpp\out\build\x64-Release ...@@ -99,4 +99,4 @@ cd /d D:\projects\PaddleSeg\deploy\cpp\out\build\x64-Release
demo.exe --conf=/path/to/your/conf --input_dir=/path/to/your/input/data/directory demo.exe --conf=/path/to/your/conf --input_dir=/path/to/your/input/data/directory
``` ```
更详细说明请参考ReadMe文档: [预测和可视化部分](../ReadMe.md) 更详细说明请参考ReadMe文档: [预测和可视化部分](../README.md)
...@@ -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.
先完成此消息的编辑!
想要评论请 注册