yys_auto/scriptBase/ocr.py

from paddleocr import PaddleOCR
from PIL import Image, ImageGrab
import paddle
import paddleocr
import numpy as np

print(paddle.__version__)  # 打印 PaddlePaddle 版本
print(paddleocr.__version__)  # 打印 PaddleOCR 版本

# 定义截图和OCR识别函数
def capture_and_ocr(region):
    # 读取屏幕截图
    screenshot = ImageGrab.grab(bbox=region)
    # 将 PIL.Image 转换为 numpy 数组
    screenshot_np = np.array(screenshot)

    ocr = PaddleOCR(use_angle_cls=True)  # 支持中文
    result = ocr.ocr(screenshot_np, cls=True)
    final_texts = [line[1][0] for line in result[0]]  # 提取每个识别结果的文本部分
    return final_texts