爬虫7 验证码解析

from PIL import Image
from pytesseract import image_to_string

# 一. 处理简单的，干扰少或无干扰的验证码图像
img = Image.open(‘timg2.jpg‘)
print(‘--------‘, img)
code = image_to_string(img)
print(code)


# 二、处理有复杂干扰的验证码图像，需根据实际情况修改代码
"""
    1. 处理图像，降噪，去除干扰，转化灰度等等，需要根据真实情况来设计
    2. 字符识别
"""


def clear_image(image):
    image = image.convert(‘RGB‘)
    width = image.size[0]
    height = image.size[1]
    noise_color = get_noise_color(image)
    
    for x in range(width):
        for y in range(height):
            # 清除边框和干扰色
            rgb = image.getpixel((x, y))
            if (x == 0 or y == 0 or x == width - 1 or y == height - 1
                    or rgb == noise_color or rgb[1] > 100):
                image.putpixel((x, y), (255, 255, 255))
    return image


def get_noise_color(image):
    for y in range(1, image.size[1] - 1):
        # 获取第2列非白的颜色
        (r, g, b) = image.getpixel((2, y))
        if r < 255 and g < 255 and b < 255:
            return (r, g, b)


image = Image.open(‘code4.png‘)
image = clear_image(image)
# 转化为灰度图
imgry = image.convert(‘L‘)
code = image_to_string(imgry)

imgry.save("imgry1.png")
with open("code.txt", "w") as f:
    print(code)
    f.write(str(code))