OpenCV tesseract not detect single digit number in an image

0
votes

enter image description here

I am using tesseract with python. It recognizes almost all of my images with 2 or more numbers or characters

I don't want to train tesseract with "only digits" because I am recognizing characters too.

But the attached image is not recognized from tessearact

1
pythonocrtesseract

1 Answers

0
votes

I think the problem is caused by that bold border. After removing that, the digit got recognized correctly.

Above is the corrected image: enter image description here

And here's the code if you are interested:

import cv2
import numpy as np
import pytesseract


def discard(image):
    image = np.uint8(image)
    _, im_label, stts, _ = cv2.connectedComponentsWithStats(image, connectivity=4)

    msk1 = np.isin(im_label, np.where(stts[:, cv2.CC_STAT_WIDTH] > 100)[0])
    msk2 = np.isin(im_label, np.where(stts[:, cv2.CC_STAT_HEIGHT] > 100)[0])

    image[(msk1 | msk2)] = 0
    return image


img = cv2.imread("check_img.jpg", 0)

# Binarization
thresh = 255 - img
ret, thresh = cv2.threshold(thresh, 5, 255, cv2.THRESH_BINARY)

# removing long connected-components
thresh = discard(thresh)

# remove noise
thresh = cv2.medianBlur(thresh, 3)

# invert again 
thresh = 255 - thresh

# showing the image
cv2.imshow("img", thresh)

# Using Tesseract OCR
custom_config = r'--oem 3 --psm 6'
text = pytesseract.image_to_string(thresh, config=custom_config)

print(text)

cv2.waitKey(0)