opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-09-24 23:33:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix the assignment of a var

Browse Source
This commit is contained in:
shaohua.zhang
2020-07-23 13:59:58 +08:00
committed by GitHub
parent cd56032479
commit af8740330b
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/program.py
+4 -3
View File
@@ -394,8 +394,9 @@ def preprocess():
startup_program = fluid.Program()
train_program = fluid.Program()
contain_det = False
if alg in ['EAST', 'DB']:
contain_det = True
train_alg_type = 'det'
else:
train_alg_type = 'rec'
return startup_program, train_program, place, config, contain_det
return startup_program, train_program, place, config, train_alg_type