From b62e6954adfab0edb4a3fcd9835f147c89111806 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Tue, 12 Apr 2022 14:00:07 +0800 Subject: [PATCH] fix rec output name error --- tools/infer/utility.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index fd2605d9..b16aecd4 100644 --- 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) -- GitLab