32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
|
import easyocr
|
|||
|
|
import cv2
|
|||
|
|
|
|||
|
|
def ocr_image(img_path):
|
|||
|
|
"""
|
|||
|
|
Выполняет OCR для изображения с номером.
|
|||
|
|
|
|||
|
|
:param img_path: Путь к изображению.
|
|||
|
|
:return: Распознанный текст.
|
|||
|
|
"""
|
|||
|
|
reader = easyocr.Reader(['en'], gpu=True) # Инициализация easyOCR
|
|||
|
|
img = cv2.imread(img_path) # Загрузка изображения
|
|||
|
|
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) # Конвертация в оттенки серого
|
|||
|
|
|
|||
|
|
# Запуск OCR
|
|||
|
|
results = reader.readtext(gray)
|
|||
|
|
print(results)
|
|||
|
|
text = ""
|
|||
|
|
for res in results:
|
|||
|
|
if len(results) == 1:
|
|||
|
|
text = res[1]
|
|||
|
|
if len(results) > 1 and len(res[1]) > 6 and res[2] > 0.2: # Фильтрация текста
|
|||
|
|
text = res[1]
|
|||
|
|
|
|||
|
|
return text
|
|||
|
|
|
|||
|
|
# Использование
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
img_path = "result.png" # Путь к изображению с номером
|
|||
|
|
recognized_text = ocr_image(img_path)
|
|||
|
|
print(f"Распознанный текст: {recognized_text}")
|