mirror of
https://github.com/RapidAI/RapidOCR.git
synced 2026-11-11 13:30:51 +08:00
21 lines
621 B
Python
21 lines
621 B
Python
# -*- encoding: utf-8 -*-
|
|
# @Author: SWHL
|
|
# @Contact: liekkaskono@163.com
|
|
from pathlib import Path
|
|
from typing import Dict, Union
|
|
|
|
import yaml
|
|
|
|
from .infer_engine import TorchInferSession
|
|
from .load_image import LoadImage, LoadImageError
|
|
from .logger import get_logger
|
|
from .parse_parameters import UpdateParameters, init_args, update_model_path
|
|
from .process_img import add_round_letterbox, increase_min_side, reduce_max_side
|
|
from .vis_res import VisRes
|
|
|
|
|
|
def read_yaml(yaml_path: Union[str, Path]) -> Dict[str, Dict]:
|
|
with open(yaml_path, "rb") as f:
|
|
data = yaml.load(f, Loader=yaml.Loader)
|
|
return data
|