tomatocream/tooling
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

a

Browse Source
This commit is contained in:
dingfeng.wong
2025-07-22 01:30:30 +08:00
parent 27b391d7fe
commit a3779aa3f9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/tooling/ocr_screenshot.py
+2 -2
View File
@@ -56,8 +56,8 @@ def perform_ocr(image_path: str, lang: str = 'eng') -> str:
Extracted text from the image Extracted text from the image
""" """
try: try:
# Load the OCR model with reliable architecture # Load the OCR model with state-of-the-art PARSeq recognition
model = ocr_predictor(det_arch='db_resnet50', reco_arch='crnn_vgg16_bn', pretrained=True) model = ocr_predictor(det_arch='db_resnet50', reco_arch='parseq', pretrained=True)
# Load the document from the image file # Load the document from the image file
doc = DocumentFile.from_images(image_path) doc = DocumentFile.from_images(image_path)