From c49d0003893115843a4d8ff0dcb9cfbf9eaab403 Mon Sep 17 00:00:00 2001 From: tink2123 Date: Wed, 17 Aug 2022 17:13:06 +0800 Subject: [PATCH] rever rec input h --- tools/export_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/export_model.py b/tools/export_model.py index a61e5ca9..e0177095 100755 --- a/tools/export_model.py +++ b/tools/export_model.py @@ -128,7 +128,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": -- GitLab