22 lines
670 B
Python
22 lines
670 B
Python
from typing import TYPE_CHECKING
|
|
|
|
from configs.basic_config import logger
|
|
|
|
# if TYPE_CHECKING:
|
|
# try:
|
|
# from rapidocr_paddle import RapidOCR
|
|
# except ImportError:
|
|
# from rapidocr_onnxruntime import RapidOCR
|
|
|
|
|
|
def get_ocr(use_cuda: bool = True) -> "RapidOCR":
|
|
try:
|
|
from rapidocr_paddle import RapidOCR
|
|
logger.info(f"from rapidocr_paddle import RapidOCR")
|
|
ocr = RapidOCR(det_use_cuda=use_cuda, cls_use_cuda=use_cuda, rec_use_cuda=use_cuda)
|
|
except ImportError:
|
|
from rapidocr_onnxruntime import RapidOCR
|
|
logger.info(f"from rapidocr_onnxruntime import RapidOCR")
|
|
ocr = RapidOCR()
|
|
return ocr
|