This commit is contained in:
Xie Xuyang 2022-09-29 11:19:44 +08:00 committed by GitHub
parent ca27d79ce2
commit f699d787ff
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -480,7 +480,8 @@ class PaddleOCR(predict_system.TextSystem):
params.rec_image_shape = "3, 48, 320" params.rec_image_shape = "3, 48, 320"
else: else:
params.rec_image_shape = "3, 32, 320" params.rec_image_shape = "3, 32, 320"
# download model # download model if using paddle infer
if not params.use_onnx:
maybe_download(params.det_model_dir, det_url) maybe_download(params.det_model_dir, det_url)
maybe_download(params.rec_model_dir, rec_url) maybe_download(params.rec_model_dir, rec_url)
maybe_download(params.cls_model_dir, cls_url) maybe_download(params.cls_model_dir, cls_url)