Как извлечь текст из изображения с помощью pytesseract? - PullRequest
Новая
       47

Как извлечь текст из изображения с помощью pytesseract?

1 голос
/ 24 июня 2019

Я использую pytesseract, чтобы попытаться извлечь текстовые номера из изображения.

Я пытаюсь извлечь три числа из этой картинки.

Простой метод с использованием pytesseract: 

from PIL import Image
from pytesseract import pytesseract
text = pytesseract.image_to_string(Image.open("uploaded_image.png"))
print(text)

Но это печатает пустым.

Почему он не может извлечь числа, как это может для обычного обычного текста?

1 Ответ

1 голос
/ 24 июня 2019

Ваши изображения нуждаются в предварительной обработке, чтобы их можно было эффективно обработать с помощью pytesseract.

Ниже показан этот процесс с использованием операций cv2.adaptiveThreshold(), cv2.findContours(), cv2.drawContours() до преобразования изображения в черно-белое и его инвертирования: 

import numpy as np
import cv2
from PIL import Image
import pytesseract

img = cv2.imread('uploaded_image.png', cv2.IMREAD_COLOR)
img = cv2.blur(img, (5, 5))

#HSV (hue, saturation, value)
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
h, s, v = cv2.split(hsv)

#Applying threshold on pixels' Value (or Brightness)
thresh = cv2.adaptiveThreshold(v, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY_INV, 11, 2)

#Finding contours
contours, hierarchy = cv2.findContours(thresh,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)

#Filling contours
contours = cv2.drawContours(img,np.array(contours),-1,(255,255,255),-1)

#To black and white
grayImage = cv2.cvtColor(contours, cv2.COLOR_BGR2GRAY)

#And inverting it
#Setting all `dark` pixels to white
grayImage[grayImage > 200] = 0
#Setting relatively clearer pixels to black
grayImage[grayImage < 100] = 255
#Write the temp file
cv2.imwrite('temp.png',grayImage)

#Read it with tesseract
text = pytesseract.image_to_string(Image.open('temp.png'),config='tessedit_char_whitelist=0123456789 -psm 6 ')

#Output
print("####  Raw text ####")
print(text)
print()
print("#### Extracted digits ####")
print([''.join([y for y in x if y.isdigit()]) for x in text.split('\n')])

Вывод 

####  Raw text ####
93
31
92

#### Extracted digits ####
['93', '31', '92']

Обработанное изображение :

enter image description here

РЕДАКТИРОВАТЬ

Обновлен ответ с использованием библиотеки cv2 и получение всех цифр из изображения

...