Skip to content

GitLab

Commit 2d14b487 authored by andyjpaddle's avatar andyjpaddle
Browse files

fix det slim export

parent b41fd6ee
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
deploy/slim/quantization/export_model.py
View file @ 2d14b487
...@@ -120,8 +120,7 @@ def main(): ...@@ -120,8 +120,7 @@ def main():
for k, v in metric.items(): for k, v in metric.items():
logger.info('{}:{}'.format(k, v)) 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"] save_path = config["Global"]["save_inference_dir"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment