fix infer

2020-04-26 05:39:35 +00:00 · 2020-04-26 05:39:35 +00:00 · 47a6f1a697
parent 73e34e7757
commit 47a6f1a697
1 changed files with 1 additions and 0 deletions
--- a/tools/infer/infer.py
+++ b/tools/infer/infer.py
@ -105,6 +105,7 @@ def main():
    exe, program, feed_names, fetch_names = 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_names,