Browse Source

Fix ocr utills

陆逊 10 months ago
parent
commit
fbf1c4bf98
1 changed files with 13 additions and 13 deletions
  1. 13 13
      magic_pdf/model/sub_modules/ocr/paddleocr/ocr_utils.py

+ 13 - 13
magic_pdf/model/sub_modules/ocr/paddleocr/ocr_utils.py

@@ -330,19 +330,19 @@ def onnx_model_init(key):
 
     import importlib.resources
 
-    resource_path = importlib.resources.path('rapidocr_onnxruntime.models','')
-
-    onnx_model = None
-    additional_ocr_params = {
-        "use_onnx": True,
-        "det_model_dir": f'{resource_path}/ch_PP-OCRv4_det_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',
-        "det_db_box_thresh": key[1],
-        "use_dilation": key[2],
-        "det_db_unclip_ratio": key[3],
-    }
-    # logger.info(f"additional_ocr_params: {additional_ocr_params}")
+    with importlib.resources.path('rapidocr_onnxruntime.models','') as resource_path:
+        onnx_model = None
+        additional_ocr_params = {
+            "use_onnx": True,
+            "det_model_dir": f'{resource_path}/ch_PP-OCRv4_det_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',
+            "det_db_box_thresh": key[1],
+            "use_dilation": key[2],
+            "det_db_unclip_ratio": key[3],
+        }
+        # logger.info(f"additional_ocr_params: {additional_ocr_params}")
+
     if key[0] is not None:
         additional_ocr_params["lang"] = key[0]