图片解析应用
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.

18 lines
797 B

  1. class CnOcr(object):
  2. def __init__(
  3. self,
  4. rec_model_name: str = 'densenet_lite_136-fc',
  5. *,
  6. det_model_name: str = 'ch_PP-OCRv3_det',
  7. cand_alphabet: Optional[Union[Collection, str]] = None,
  8. context: str = 'cpu', # ['cpu', 'gpu', 'cuda']
  9. rec_model_fp: Optional[str] = None,
  10. rec_model_backend: str = 'onnx', # ['pytorch', 'onnx']
  11. rec_vocab_fp: Union[str, Path] = VOCAB_FP,
  12. rec_more_configs: Optional[Dict[str, Any]] = None,
  13. rec_root: Union[str, Path] = data_dir(),
  14. det_model_fp: Optional[str] = None,
  15. det_model_backend: str = 'onnx', # ['pytorch', 'onnx']
  16. det_more_configs: Optional[Dict[str, Any]] = None,
  17. det_root: Union[str, Path] = det_data_dir(),
  18. **kwargs,
  19. )