Skip to content

P
PaddleClas

PaddlePaddle / PaddleClas
大约 1 年 前同步成功

通知 115
Star 4999
Fork 1114
P
PaddleClas
提交 77450190 编写于 作者: littletomatodonkey's avatar littletomatodonkey
浏览文件
操作

fix pyinfer

上级 47a6f1a6
隐藏空白更改
内联 并排
Showing with 1 addition and 0 deletion
tools/infer/py_infer.py
浏览文件 @ 77450190
...@@ -87,6 +87,7 @@ def main(): ...@@ -87,6 +87,7 @@ def main():
exe, program, feed_names, fetch_lists = create_predictor(args) exe, program, feed_names, fetch_lists = create_predictor(args)
data = preprocess(args.image_file, operators) data = preprocess(args.image_file, operators)
data = np.expand_dims(data, axis=0)
outputs = exe.run(program, outputs = exe.run(program,
feed={feed_names[0]: data}, feed={feed_names[0]: data},
fetch_list=fetch_lists, fetch_list=fetch_lists,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或