add distillation function

2026-09-24 23:33:08 +08:00 · 2021-06-02 08:31:57 +00:00
parent 551a6827f0
commit ed02b91d26
17 changed files with 407 additions and 81 deletions
@@ -386,7 +386,7 @@ def preprocess(is_train=False):
    alg = config['Architecture']['algorithm']
    assert alg in [
        'EAST', 'DB', 'SAST', 'Rosetta', 'CRNN', 'STARNet', 'RARE', 'SRN',
-        'CLS', 'PGNet'
+        'CLS', 'PGNet', 'Distillation'
    ]

    device = 'gpu:{}'.format(dist.ParallelEnv().dev_id) if use_gpu else 'cpu'