简单验证码识别

对较简单的验证码进行识别并返回结果
含：数字，英文

"""
    1、将验证码进行二值化处理，保存到本地
    2、放大验证码的尺寸，保存到本地
    3、识别验证码，返回结果
"""

from PIL import Image
from PIL import ImageFile
import tesserocr


class ProcessImg:
    def __init__(self):
        self.filename = 'yanzhengma.png'
        self.width = 200
        self.height = 100
        self.type = 'png'
        self.i = 'in.png'
        self.o = 'out.png'

    def get_grey_img(self, filename):
        """
        :param filename: 'yanzhengma.png'
        :return: 'result.png'
        """
        image = Image.open(filename)
        image = image.convert('L')
        threshold = 130
        table = []
        for i in range(256):
            if i < threshold:
                table.append(0)
            else:
                table.append(1)

        image = image.point(table, '1')
        # show:弹窗查看，调整阈值 threshold
        # image.show()
        image.save('in.png')

    def ResizeImage(self, filein, fileout, width, height, type):
        """
        :param filein: 输入图片
        :param fileout: 输出图片
        :param  输出图片宽度
        :param height: 输出图片高度
        :param type: 输出图片类型（png, gif, jpeg...）
        :return:
        """
        img = Image.open(filein)
        out = img.resize((width, height), Image.ANTIALIAS)
        out.save(fileout, type)

    def change_img_size(self, width, height, type, i, o):
        ImageFile.LOAD_TRUNCATED_IMAGES = True

        self.ResizeImage(i, o, width, height, type)

    def get_result(self):
        return tesserocr.file_to_text('out.png')

    def run(self):
        """

        :return: 识别结果 str
        """
        self.get_grey_img(self.filename)
        self.change_img_size(self.width, self.height, self.type, self.i, self.o)
        return self.get_result()


if __name__ == '__main__':
    start = ProcessImg()
    result = start.run()