yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-11-30 17:05:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix SLANeXt export bug (#14512)

Browse Source 
* add slanext models

* refine codes

* refine codes

* refine codes

* fix export SLANeXt
This commit is contained in:
Liu Jiaxuan 2025-01-07 19:21:34 +08:00 committed by GitHub
parent 359ab6cb76
commit a6b96bbfb1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ppocr/utils/export_model.py
View File

@ -241,6 +241,14 @@ def dynamic_to_static(model, arch_config, logger, input_shape=None):
],
full_graph=True,
)
elif arch_config["algorithm"] == "SLANeXt":
model = paddle.jit.to_static(
model,
input_spec=[
paddle.static.InputSpec(shape=[-1, 1, 512, 512], dtype="float32")
],
full_graph=True,
)
elif arch_config["algorithm"] == "PP-FormulaNet-L":
model = paddle.jit.to_static(
model,