yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-12-27 15:08:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support variable length input

Browse Source
This commit is contained in:
WenmuZhou 2022-08-15 09:00:35 +00:00
parent ebe3e885c0
commit 7efee821bc
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/export_model.py
View File

@ -140,6 +140,8 @@ def export_single_model(model,
infer_shape = [3, 488, 488]
if arch_config["algorithm"] == "TableMaster":
infer_shape = [3, 480, 480]
if arch_config["algorithm"] == "SLANet":
infer_shape = [3, -1, -1]
model = to_static(
model,
input_spec=[