From 2d14b487ca24b1c82e683b2cc3bbc536d88ea3d1 Mon Sep 17 00:00:00 2001 From: andyjpaddle Date: Thu, 9 Dec 2021 06:36:45 +0000 Subject: [PATCH] fix det slim export --- deploy/slim/quantization/export_model.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/deploy/slim/quantization/export_model.py b/deploy/slim/quantization/export_model.py index 9080feea74..7c3abb3086 100755 --- a/deploy/slim/quantization/export_model.py +++ b/deploy/slim/quantization/export_model.py @@ -120,8 +120,7 @@ def main(): for k, v in metric.items(): logger.info('{}:{}'.format(k, v)) - infer_shape = [3, 32, 100] if config['Architecture'][ - 'model_type'] != "det" else [3, 640, 640] + infer_shape = [3, 32, 100] if model_type != "det" else [3, 640, 640] save_path = config["Global"]["save_inference_dir"]