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

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

Fix some bugs
......@@ -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"
......
......@@ -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
```
更详细说明请参考ReadMe文档: [预测和可视化部分](../ReadMe.md)
更详细说明请参考ReadMe文档: [预测和可视化部分](../README.md)
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册