zhao/pepperbot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

改进PaddleOCR初始化

Browse source
This commit is contained in:
zhbaor 2023-08-13 20:05:54 +08:00
parent efa1b3bd3d
commit 4a1d3978ad
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
apps/riic_report_analysis.py
View file

@ -3,6 +3,9 @@ from paddleocr import PaddleOCR
import os import os
ocr = PaddleOCR(use_angle_cls=False, use_gpu=False)
class RIICReportAnalysis: class RIICReportAnalysis:
async def group_message(self, chain: MessageChain): async def group_message(self, chain: MessageChain):
if len(chain) != 1 or len(chain.images) != 1: if len(chain) != 1 or len(chain.images) != 1:
@ -11,8 +14,7 @@ class RIICReportAnalysis:
img_path = await img_seg.download() img_path = await img_seg.download()
print(f"Image saved to {img_path}") print(f"Image saved to {img_path}")
ocr = PaddleOCR(use_angle_cls=False, lang="ch", use_gpu=False) ocr_text = ocr.ocr(img_path, cls=False)
ocr_text = ocr.ocr(img_path, det=False)
print(ocr_text) print(ocr_text)
os.remove(img_path) os.remove(img_path)