16 lines
469 B
Python
16 lines
469 B
Python
from rapidocr_onnxruntime import RapidOCR
|
|
|
|
from mower.utils import typealias as tp
|
|
from mower.utils.log import logger
|
|
|
|
logger.info("加载OCR")
|
|
engine = RapidOCR(text_score=0.3)
|
|
logger.info("OCR加载完成")
|
|
|
|
|
|
def ocr_rec(img: tp.Image | tp.GrayImage) -> str | None:
|
|
result, elapse = engine(img, use_det=False, use_cls=False, use_rec=True)
|
|
elapse = round(elapse[0] * 1000)
|
|
text = result[0][0].strip()
|
|
logger.debug(f"{elapse}ms {text}")
|
|
return text
|