fix det slim export

2d14b487 · andyjpaddle · b41fd6ee · 2d14b487
显示空白变更内容
内联并排

Showing with 1 addition and 2 deletion

deploy/slim/quantization/export_model.py deploy/slim/quantization/export_model.py +1 -2

未找到文件。
--- 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'][
+    infer_shape = [3, 32, 100] if model_type != "det" else [3, 640, 640]
-        'model_type'] != "det" else [3, 640, 640]
    save_path = config["Global"]["save_inference_dir"]