提交 77450190 编写于 作者: littletomatodonkey's avatar littletomatodonkey

fix pyinfer

上级 47a6f1a6
......@@ -87,6 +87,7 @@ def main():
exe, program, feed_names, fetch_lists = create_predictor(args)
data = preprocess(args.image_file, operators)
data = np.expand_dims(data, axis=0)
outputs = exe.run(program,
feed={feed_names[0]: data},
fetch_list=fetch_lists,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册