yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-11-02 10:49:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update program.py

Browse Source
This commit is contained in:
dyning 2020-05-15 22:25:37 +08:00 committed by GitHub
parent 28ab6a3b4b
commit f5fb36e88c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/program.py
View File

@ -191,8 +191,8 @@ def build_export(config, main_prog, startup_prog):
func_infor = config['Architecture']['function']
model = create_module(func_infor)(params=config)
image, outputs = model(mode='export')
fetches_var = sorted([outputs[name] for name in outputs])
fetches_var_name = [name for name in fetches_var]
fetches_var_name = sorted([name for name in outputs])
fetches_var = [outputs[name] for name in fetches_var_name]
feeded_var_names = [image.name]
target_vars = fetches_var
return feeded_var_names, target_vars, fetches_var_name