forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix download model error

This commit is contained in:
WenmuZhou 2021-06-08 14:31:42 +08:00
parent d27360a9b8
commit 2f62c953e2
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ppstructure/paddlestructure.py
View File

@ -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)))