Skip to content

GitLab

"tests/git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "80ca01bdd1d6e737df405cefb67777852d3100cf"
Commit fbf1c4bf authored by 陆逊's avatar 陆逊
Browse files

Fix ocr utills

parent 69273e96
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 @ fbf1c4bf
...@@ -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