You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
797 B
19 lines
797 B
class CnOcr(object):
|
|
def __init__(
|
|
self,
|
|
rec_model_name: str = 'densenet_lite_136-fc',
|
|
*,
|
|
det_model_name: str = 'ch_PP-OCRv3_det',
|
|
cand_alphabet: Optional[Union[Collection, str]] = None,
|
|
context: str = 'cpu', # ['cpu', 'gpu', 'cuda']
|
|
rec_model_fp: Optional[str] = None,
|
|
rec_model_backend: str = 'onnx', # ['pytorch', 'onnx']
|
|
rec_vocab_fp: Union[str, Path] = VOCAB_FP,
|
|
rec_more_configs: Optional[Dict[str, Any]] = None,
|
|
rec_root: Union[str, Path] = data_dir(),
|
|
det_model_fp: Optional[str] = None,
|
|
det_model_backend: str = 'onnx', # ['pytorch', 'onnx']
|
|
det_more_configs: Optional[Dict[str, Any]] = None,
|
|
det_root: Union[str, Path] = det_data_dir(),
|
|
**kwargs,
|
|
)
|