Merge pull request #7232 from tink2123/revert_imgh_2.5

rever rec input h
2022-08-17 17:16:01 +08:00 · 2022-08-17 17:16:01 +08:00 · eed3d6aaa2
parent 93b4954136 0b97b0c4a3
commit eed3d6aaa2
1 changed files with 1 additions and 1 deletions
--- a/tools/export_model.py
+++ b/tools/export_model.py
@ -83,7 +83,7 @@ def export_single_model(model,
    else:
        infer_shape = [3, -1, -1]
        if arch_config["model_type"] == "rec":
-            infer_shape = [3, 48, -1]  # for rec model, H must be 32
+            infer_shape = [3, 32, -1]  # for rec model, H must be 32
            if "Transform" in arch_config and arch_config[
                    "Transform"] is not None and arch_config["Transform"][
                        "name"] == "TPS":