ocr.py 528 B

12345678910111213141516
  1. from paddleocr import PaddleOCR
  2. from PIL import Image, ImageGrab
  3. import numpy as np
  4. # 定义截图和OCR识别函数
  5. def capture_and_ocr(region):
  6. # 读取屏幕截图
  7. screenshot = ImageGrab.grab(bbox=region)
  8. # 将 PIL.Image 转换为 numpy 数组
  9. screenshot_np = np.array(screenshot)
  10. ocr = PaddleOCR(use_angle_cls=True) # 支持中文
  11. result = ocr.ocr(screenshot_np, cls=True)
  12. final_texts = [line[1][0] for line in result[0]] # 提取每个识别结果的文本部分
  13. return final_texts