[Feature] Add PREN Scene Text Recognition Model(Accepted in CVPR2021) (#5563)

* [Feature] add PREN scene text recognition model

* [Patch] Optimize yml File

* [Patch] Save Label/Pred Preprocess Time Cost

* [BugFix] Modify Shape Conversion to Fit for Inference Model Exportion

* [Patch] ?

* [Patch] ?

* 啥情况...
This commit is contained in:
OneYearIsEnough
2022-02-28 21:48:00 +08:00
committed by GitHub
parent 3df64d502b
commit 6e607a0fa1
17 changed files with 698 additions and 11 deletions
+1 -1
View File
@@ -541,7 +541,7 @@ def preprocess(is_train=False):
assert alg in [
'EAST', 'DB', 'SAST', 'Rosetta', 'CRNN', 'STARNet', 'RARE', 'SRN',
'CLS', 'PGNet', 'Distillation', 'NRTR', 'TableAttn', 'SAR', 'PSE',
'SEED', 'SDMGR', 'LayoutXLM', 'LayoutLM'
'SEED', 'SDMGR', 'LayoutXLM', 'LayoutLM', 'PREN'
]
device = 'cpu'