aho/welcomecenterbot
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

normalized-log-fix10

Browse Source
This commit is contained in:
Untone 2024-09-28 11:33:55 +03:00
parent 494cc5766f
commit d9e9c547ef
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
nlp/ocr.py
View File

@ -7,12 +7,15 @@ logger = logging.getLogger(" ocr ")
reader = easyocr.Reader(['ru']) # Specify the languages you want to support reader = easyocr.Reader(['ru']) # Specify the languages you want to support
def ocr_recognize(file_path): def ocr_recognize(file_path):
sum_text = ""
# Use EasyOCR to detect text in the photo # Use EasyOCR to detect text in the photo
result = reader.readtext(file_path) results = reader.readtext(file_path)
logger.debug("OCR Result: %s", result) for result in results:
[_coords, ocr_text, ocr_accuracy] = result
logger.debug("OCR Result: %s", ocr_text)
if ocr_accuracy.item() > 0.5:
sum_text += ocr_text
# Extract recognized text logger.debug(f'Recognized Text: {sum_text}')
recognized_text = ' '.join([text[1] for text in result if isinstance(text, tuple) and len(text) > 1]) return sum_text
logger.debug(f'Recognized Text: {recognized_text}')
return recognized_text