调用PaddleOCR识别文字
This commit is contained in:
parent
18372e81c6
commit
8f5d31725e
1 changed files with 15 additions and 3 deletions
|
@ -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:
|
||||
async def group_message(self, chain: MessageChain):
|
||||
if chain.pure_text == "芜湖":
|
||||
await chain.onebot_reply(Text("起飞"))
|
||||
if len(chain) != 1:
|
||||
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.")
|
||||
|
|
Loading…
Reference in a new issue