mirror of
https://github.com/RapidAI/RapidOCR.git
synced 2026-11-11 21:40:52 +08:00
31 lines
954 B
Python
31 lines
954 B
Python
# -*- encoding: utf-8 -*-
|
|
# @Author: SWHL
|
|
# @Contact: liekkaskono@163.com
|
|
import importlib
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import yaml
|
|
|
|
|
|
class BaseModule(object):
|
|
def __init__(self, package_name: str = 'rapidocr_onnxruntime'):
|
|
self.package_name = package_name
|
|
self.root_dir = Path(__file__).resolve().parent.parent
|
|
|
|
sys.path.append(str(self.root_dir))
|
|
sys.path.append(str(self.root_dir / self.package_name))
|
|
|
|
def init_module(self, module_name: str, class_name: str = None):
|
|
if class_name is None:
|
|
module_part = importlib.import_module(f'{self.package_name}')
|
|
return module_part
|
|
module_part = importlib.import_module(f'{self.package_name}.{module_name}')
|
|
return getattr(module_part, class_name)
|
|
|
|
@staticmethod
|
|
def read_yaml(yaml_path: str):
|
|
with open(yaml_path, 'rb') as f:
|
|
data = yaml.load(f, Loader=yaml.Loader)
|
|
return data
|