mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2026-09-01 15:32:24 +08:00
change support list (#14293)
This commit is contained in:
@@ -41,7 +41,16 @@ def apply_to_static(model, config, logger):
|
||||
assert (
|
||||
"d2s_train_image_shape" in config["Global"]
|
||||
), "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"]:
|
||||
algo = list(config["Architecture"]["Models"].values())[0]["algorithm"]
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user