Merge branch 'dygraph' into dygraph_rc

52f45bc3 · zhoujun · GitHub · f5315349 · f2fc1a3f · 52f45bc3
Unverified Commit 52f45bc3 authored Apr 09, 2021 by zhoujun Committed by GitHub Apr 09, 2021
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 124 additions and 1 deletion

tools/infer_e2e.py tools/infer_e2e.py +122 -0

tools/program.py tools/program.py +2 -1

No files found.
--- a/tools/infer_e2e.py
+++ b/tools/infer_e2e.py
--- a/tools/program.py
+++ b/tools/program.py
@@ -375,7 +375,8 @@ def preprocess(is_train=False):

    alg = config['Architecture']['algorithm']
    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'