ocr/main.py

15 lines
332 B
Python
Raw Permalink Normal View History

2024-01-22 13:09:42 +08:00
from PIL import Image
2024-01-22 12:29:28 +08:00
from paddleocr import PaddleOCR
# 加载模型 (en ch japan korean ru)
print(f'开始加载模型...')
CH = PaddleOCR(use_angle_cls=True, lang="ch")
2024-01-22 13:09:42 +08:00
# 加载图像
2024-01-22 12:29:28 +08:00
print(f'开始加载图像...')
2024-01-22 13:09:42 +08:00
image = Image.open('./test.jpg')
2024-01-22 12:29:28 +08:00
print(f'开始提取文字...')
2024-01-22 13:09:42 +08:00
data = CH.ocr(image, cls=True)[0]
print(data)