浏览代码

feat(ocr): update OCR utility and dependencies

- Update the default configuration path in pytorchocr_utility.py
- Add required dependencies for paddleocr2pytorch in setup.py:
  - shapely
  - pyclipper
  - omegaconf
myhloli 7 月之前
父节点
当前提交
d09464be06

+ 0 - 0
magic_pdf/model/sub_modules/ocr/paddleocr2pytorch/pytorchocr/modeling/arch_config.yaml → magic_pdf/model/sub_modules/ocr/paddleocr2pytorch/pytorchocr/utils/resources/arch_config.yaml


+ 1 - 1
magic_pdf/model/sub_modules/ocr/paddleocr2pytorch/tools/infer/pytorchocr_utility.py

@@ -7,7 +7,7 @@ import argparse
 
 
 root_dir = Path(__file__).resolve().parent.parent.parent
-DEFAULT_CFG_PATH = root_dir / "pytorchocr" / "modeling" / "arch_config.yaml"
+DEFAULT_CFG_PATH = root_dir / "pytorchocr" / "utils" / "resources" / "arch_config.yaml"
 
 
 def init_args():

+ 3 - 0
setup.py

@@ -47,6 +47,9 @@ if __name__ == '__main__':
                      "PyYAML",  # yaml
                      "ftfy", # unimernet_hf
                      "openai",  # openai SDK
+                     "shapely",  # imgaug-paddleocr2pytorch
+                     "pyclipper",  # paddleocr2pytorch
+                     "omegaconf",  # paddleocr2pytorch
                      ],
             "old_linux":[
                 "albumentations<=1.4.20", # 1.4.21引入的simsimd不支持2019年及更早的linux系统