fix pyinfer

2020-04-26 05:49:49 +00:00 · 2020-04-26 05:49:49 +00:00 · 77450190af
parent 47a6f1a697
commit 77450190af
1 changed files with 1 additions and 0 deletions
--- a/tools/infer/py_infer.py
+++ b/tools/infer/py_infer.py
@ -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,