Exploring OCR with OpenCV and PyTesseract

The utility of OCR extends beyond its applications in various domains, from document digitization to text extraction in images. In this blog, we are exploring the performance and reliability of OCR using OpenCV and PyTesseract on a diverse set of images.

The blog comprises two key phases:

1. The first focuses on extracting text from grayscale images
2. The second is dedicated to detecting and extracting text from color images

The results will shed light on the accuracy and challenges associated with OCR, providing insights into its potential applications and limitations.

Experimental Setup

Step 1: Library Installation

Before commencing the experiment, we installed two crucial libraries:

• OpenCV (Open Source Computer Vision Library): An open-source library specializing in computer vision and machine learning tasks, including image processing and object detection.
• Python-tesseract (Pytesseract): An optical character recognition (OCR) tool in Python known for its ability to extract text from images.

Here’s how we go about it:

# To install opencv
pip install opencv-python

# To install pytesseract
pip install pytesseract

Step 2: OCR Text extraction from a Grayscale Image

Our journey commences with the extraction of text from a grayscale image. We’ll begin by loading an input image from which we intend to extract text.

Import Libraries

from PIL import Image
from pytesseract import pytesseract

Reading and Resizing the Image

We read the image and resize it to the desired dimensions.

Note: If you wish to save the resized image, you can use the Image.save() method.

image = Image.open('ocr.png')
image = image.resize((400,200))
image.save('resized_image.png') # optional

Extracting Text

We employ the image_to_string method from the Pytesseract class to extract text from the image.

text = pytesseract.image_to_string(image)
#print the text
print('detected text : ',text)

Here is what the output looks like :

Step 3: Detecting and Extracting Text from Color Images

Now, we venture into the process of extracting text from color images. Take a look at the example below, showcasing the color image from which we’ll be extracting text:

Here, we will draw rectangular bounding boxes around the text using OpenCV.

Importing Libraries

import cv2
from pytesseract import pytesseract

Image Preprocessing

We read the image and converted it to grayscale using cv2.cvtColor.

img = cv2.imread("img_colour.jpg")
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

We then convert the grayscale image into a binary image. Binary images have only two possible pixel values, often 0 for black and 1 (or 255) for white. This simplifies the information and is typically achieved through thresholding, a technique for distinguishing the foreground from the background.

ret, thresh1 = cv2.threshold(gray, 0, 255, cv2.THRESH_OTSU |
                                          cv2.THRESH_BINARY_INV)
cv2.imwrite('img_thresholding.jpg',thresh1)

Step 4: Bounding Boxes and Text Extraction

We define a rectangular kernel using cv2.getStructuringElement in OpenCV.

In this function, the first argument is the grayscale image, and the second argument is our threshold value T, which we’ve set to 0. That’s because Otsu’s method automatically calculates our optimal threshold value. The third argument is the output value when a pixel passes the threshold test.

The fourth argument is the thresholding type, which is logically combined with two methods.

The cv2.threshold function returns a tuple of two values: the threshold value T and the thresholded image itself.

We the create a rectangular kernel with OpenCV’s cv2.getStructuringElement function. In OpenCV, you have the option to use either the cv2.getStructuringElement function or NumPy to define your structuring element.

rect_kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (12, 12))

Dilation acts like a magnifying glass for important parts of the image, making them larger. This helps connect broken text together, especially in challenging cases. We achieve this using the cv2.dilate function, which helps define text boundaries.

dilation = cv2.dilate(thresh1, rect_kernel, iterations = 3)
cv2.imwrite('dilation_image.jpg',dilation)

Step 5: Text Detection and Cropping

We use the cv2.findContours method to identify the areas covered by white pixels in the image.

contours, hierarchy = cv2.findContours(dilation, cv2.RETR_EXTERNAL,
                                            cv2.CHAIN_APPROX_NONE)

We then draw bounding boxes around each of these areas, helping us isolate and focus on each block of text. With the bounding boxes in place, we crop out these rectangular sections, making text extraction using Pytesseract more manageable.

for cnt in contours:
    x, y, w, h = cv2.boundingRect(cnt)
    # Draw the bounding box on the text area
    rect=cv2.rectangle(im2, (x, y), (x + w, y + h), (0, 255, 0), 2)
    # Crop the bounding box area
    cropped = im2[y:y + h, x:x + w]
    cv2.imwrite('rectanglebox.jpg',rect) #optional
    # open the text file
    file = open("text_output2.txt", "a")
    # Using tesseract on the cropped image area to get text
    text = pytesseract.image_to_string(cropped)
    # Adding the text to the file
    file.write(text)
    file.write("\\n")
    # Closing the file
    file.close

Final Words :

The game-changer lies in harnessing the synergy of deep learning with OCR, which can profoundly enhance OCR accuracy, even when dealing with diverse fonts. The latest release of Tesseract introduces deep learning-based OCR, a significant leap in accuracy driven by LSTM and RNNs.

Do You Know What is Detectron 2.

Learn more about top-10-python-libraries