mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

change support list (#14293)

pull/14297/head
liuhongen1234567 2024-11-29 15:18:54 +08:00 committed by GitHub
parent b6bcde143d
commit 8fdc409edf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ppocr/modeling/architectures/__init__.py
View File

@ -41,7 +41,16 @@ def apply_to_static(model, config, logger):
assert ( assert (
"d2s_train_image_shape" in config["Global"] "d2s_train_image_shape" in config["Global"]
), "d2s_train_image_shape must be assigned for static training mode..." ), "d2s_train_image_shape must be assigned for static training mode..."
supported_list = ["DB", "SVTR_LCNet", "TableMaster", "LayoutXLM", "SLANet", "SVTR"] supported_list = [
"DB",
"SVTR_LCNet",
"TableMaster",
"LayoutXLM",
"SLANet",
"SVTR",
"SVTR_HGNet",
"LaTeXOCR",
]
if config["Architecture"]["algorithm"] in ["Distillation"]: if config["Architecture"]["algorithm"] in ["Distillation"]:
algo = list(config["Architecture"]["Models"].values())[0]["algorithm"] algo = list(config["Architecture"]["Models"].values())[0]["algorithm"]
else: else: