Skip to content

GitLab

Unverified Commit 52f45bc3 authored by zhoujun's avatar zhoujun Committed by GitHub
Browse files

Merge branch 'dygraph' into dygraph_rc

parents f5315349 f2fc1a3f
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 124 additions and 1 deletion
tools/infer_e2e.py 0 → 100755
View file @ 52f45bc3
This diff is collapsed.
tools/program.py
View file @ 52f45bc3
...@@ -375,7 +375,8 @@ def preprocess(is_train=False): ...@@ -375,7 +375,8 @@ def preprocess(is_train=False):
alg = config['Architecture']['algorithm'] alg = config['Architecture']['algorithm']
assert alg in [ assert alg in [
'EAST', 'DB', 'SAST', 'Rosetta', 'CRNN', 'STARNet', 'RARE', 'SRN', 'CLS' 'EAST', 'DB', 'SAST', 'Rosetta', 'CRNN', 'STARNet', 'RARE', 'SRN',
'CLS', 'PGNet'
] ]
device = 'gpu:{}'.format(dist.ParallelEnv().dev_id) if use_gpu else 'cpu' device = 'gpu:{}'.format(dist.ParallelEnv().dev_id) if use_gpu else 'cpu'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment