调用PaddleOCR识别文字

This commit is contained in:
zhbaor 2023-08-13 19:42:17 +08:00
parent 18372e81c6
commit 8f5d31725e

View file

@ -1,7 +1,19 @@
from pepperbot.core.message.chain import MessageChain, Text from pepperbot.core.message.chain import MessageChain, Image, Text
from paddleocr import PaddleOCR
import os
class RIICReportAnalysis: class RIICReportAnalysis:
async def group_message(self, chain: MessageChain): async def group_message(self, chain: MessageChain):
if chain.pure_text == "芜湖": if len(chain) != 1:
await chain.onebot_reply(Text("起飞")) return
img_seg: Image = chain[0]
img_path = await img_seg.download()
print(f"Image saved to {img_path}")
ocr = PaddleOCR(use_angle_cls=False, lang="ch", use_gpu=False)
ocr_text = ocr.ocr(img_path, det=False)
print(ocr_text)
os.remove(img_path)
print(f"file removed.")