Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "e5d117c8bff0a7d6fcae58342e8b710dc9a7e594"
Unverified Commit 96d5bc07 authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Merge pull request #1578 from moria97/fix-ocr 

LGTM
parents 48a43370 fbf1c4bf
Hide whitespace changes
Inline Side-by-side
Showing with 13 additions and 13 deletions
magic_pdf/model/sub_modules/ocr/paddleocr/ocr_utils.py
View file @ 96d5bc07
...@@ -330,19 +330,19 @@ def onnx_model_init(key): ...@@ -330,19 +330,19 @@ def onnx_model_init(key):
import importlib.resources import importlib.resources
resource_path = importlib.resources.path('rapidocr_onnxruntime.models','') with importlib.resources.path('rapidocr_onnxruntime.models','') as resource_path:
onnx_model = None
onnx_model = None additional_ocr_params = {
additional_ocr_params = { "use_onnx": True,
"use_onnx": True, "det_model_dir": f'{resource_path}/ch_PP-OCRv4_det_infer.onnx',
"det_model_dir": f'{resource_path}/ch_PP-OCRv4_det_infer.onnx', "rec_model_dir": f'{resource_path}/ch_PP-OCRv4_rec_infer.onnx',
"rec_model_dir": f'{resource_path}/ch_PP-OCRv4_rec_infer.onnx', "cls_model_dir": f'{resource_path}/ch_ppocr_mobile_v2.0_cls_infer.onnx',
"cls_model_dir": f'{resource_path}/ch_ppocr_mobile_v2.0_cls_infer.onnx', "det_db_box_thresh": key[1],
"det_db_box_thresh": key[1], "use_dilation": key[2],
"use_dilation": key[2], "det_db_unclip_ratio": key[3],
"det_db_unclip_ratio": key[3], }
} # logger.info(f"additional_ocr_params: {additional_ocr_params}")
# logger.info(f"additional_ocr_params: {additional_ocr_params}")
if key[0] is not None: if key[0] is not None:
additional_ocr_params["lang"] = key[0] additional_ocr_params["lang"] = key[0]
......
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