opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-10-25 10:17:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #5950 from LDOUBLEV/dygraph

Browse Source 
fix rec output name error
This commit is contained in:
andyjpaddle
2022-04-12 14:58:42 +08:00
committed by GitHub
parent a8318353be 243f89da21
commit e4ab0ebe86
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/infer/utility.py
+4
View File
@@ -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)