Skip to content

M
models

PaddlePaddle / models
大约 2 年 前同步成功

通知 232
Star 6828
Fork 2962
M
models
未验证 提交 a0111439 编写于 作者: R ruri 提交者： GitHub
浏览文件
操作

fix infer, add softmax (#2330)

上级 2ae83f08
隐藏空白更改
内联 并排
Showing with 1 addition and 0 deletion
PaddleCV/image_classification/infer.py
浏览文件 @ a0111439
...@@ -47,6 +47,7 @@ def infer(args): ...@@ -47,6 +47,7 @@ def infer(args):
out, _, _ = model.net(input=image, class_dim=class_dim) out, _, _ = model.net(input=image, class_dim=class_dim)
else: else:
out = model.net(input=image, class_dim=class_dim) out = model.net(input=image, class_dim=class_dim)
out = fluid.layers.softmax(out)
test_program = fluid.default_main_program().clone(for_test=True) test_program = fluid.default_main_program().clone(for_test=True)
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或