未验证 提交 13f7b1c8 编写于 作者: A andyjpaddle 提交者: GitHub

Merge pull request #7231 from tink2123/revert_imgh

rever rec input h
...@@ -130,7 +130,7 @@ def export_single_model(model, ...@@ -130,7 +130,7 @@ def export_single_model(model,
else: else:
infer_shape = [3, -1, -1] infer_shape = [3, -1, -1]
if arch_config["model_type"] == "rec": 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[ if "Transform" in arch_config and arch_config[
"Transform"] is not None and arch_config["Transform"][ "Transform"] is not None and arch_config["Transform"][
"name"] == "TPS": "name"] == "TPS":
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册