CODEWITHSUNDEEP
pythonopencvimage-processingocrcaptcha
basskiller
basskiller

Reputation: 85

Python image clean to solve the captcha

I need to resolve captcha automatically to grab the public data from sites.

I use python and opencv. I'm newbee in solving the images processing. After search, as a method to resolve captcha I came up with next. As the text in Captha uses group of related colours I try to use the HSV format and mask, then convert image to Grayscale and use Threshold (Adaptive_THRESH_MEAN_C) to remove noise from the image.

But this is not enough to remove noise and provide automatic text recognition with OCR (Tesseract). See images below.

Is there something I can improve in my solution or there is a better way?

Original images:

enter image description here

Processed images:

enter image description here

from PIL import Image
image = Image.open("captcha-img.png").convert("L")
pixel_matrix = image.load()

# thresholding
for column in range(0, image.height):
    for row in range(0, image.width):
        if pixel_matrix[row, column] != 0:
            pixel_matrix[row, column] = 255

# stray line and pixel removal
for column in range(1, image.height - 1):
    for row in range(1, image.width - 1):
        if pixel_matrix[row, column] == 0 \
            and pixel_matrix[row, column - 1] == 255 and pixel_matrix[row, column + 1] == 255:
            pixel_matrix[row, column] = 255
        if pixel_matrix[row, column] == 0 \
            and pixel_matrix[row - 1, column] == 255 and pixel_matrix[row + 1, column] == 255:
            pixel_matrix[row, column] = 255

image.save("output.png")

Upvotes: 0

Views: 2681

Answers (1)

Christian Karcher
Christian Karcher

Reputation: 3666

If you squint your eyes, the noise gets reduced.

Same could be applied to the image by blurring it via a convolution.

Using a multi-blur approach, one can get acceptable results. Wether or not its suitable for your OCR, I do not know.

enter image description here

import numpy as np
from PIL import Image
from matplotlib import pyplot as plt
from scipy.signal import convolve2d

# get numpy array from img
image = Image.open("captcha-img.png").convert("L")
img = np.array(image)

# blur img via convolution
blur_size = 3
blur_repeat = 6
blur_kernel = np.ones((blur_size, blur_size))
for _ in range(blur_repeat):
    img = convolve2d(img, blur_kernel)

# threshold and crop
remaining_img = img > np.median(img)
border = blur_size*blur_repeat
plt.imshow(remaining_img[border:-border+1, border:-border+1])
plt.show()

Upvotes: 2

Related Questions