yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-09-25 16:15:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix det slim export

Browse Source
This commit is contained in:
andyjpaddle 2021-12-09 06:36:45 +00:00
parent b41fd6ee7c
commit 2d14b487ca
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
deploy/slim/quantization/export_model.py
View File

@ -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"]