From 2f62c953e26e7fab369c5ce2e16068d444f96077 Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Tue, 8 Jun 2021 14:31:42 +0800 Subject: [PATCH] fix download model error --- ppstructure/paddlestructure.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/ppstructure/paddlestructure.py b/ppstructure/paddlestructure.py index 686267e1..03f90565 100644 --- a/ppstructure/paddlestructure.py +++ b/ppstructure/paddlestructure.py @@ -78,8 +78,8 @@ class PaddleStructure(OCRSystem): params.structure_model_dir = os.path.join(BASE_DIR, VERSION, 'structure') # download model maybe_download(params.det_model_dir, model_urls['det']) - maybe_download(params.det_model_dir, model_urls['rec']) - maybe_download(params.det_model_dir, model_urls['structure']) + maybe_download(params.rec_model_dir, model_urls['rec']) + maybe_download(params.structure_model_dir, model_urls['structure']) if params.rec_char_dict_path is None: params.rec_char_type = 'EN' @@ -143,4 +143,3 @@ def main(): logger.info(item['res']) save_res(result, save_folder, img_name) logger.info('result save to {}'.format(os.path.join(save_folder, img_name))) -