fix rec output name error

2022-04-12 14:00:07 +08:00 · 2022-04-12 14:00:07 +08:00 · b62e6954ad
parent 0b22034119
commit b62e6954ad
1 changed files with 4 additions and 0 deletions
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@ -323,6 +323,10 @@ def get_output_tensors(args, mode, predictor):
        output_name = 'softmax_0.tmp_0'
        if output_name in output_names:
            return [predictor.get_output_handle(output_name)]
+        else:
+            for output_name in output_names:
+                output_tensor = predictor.get_output_handle(output_name)
+                output_tensors.append(output_tensor)
    else:
        for output_name in output_names:
            output_tensor = predictor.get_output_handle(output_name)