验证码生成与识别

原

验证码生成与识别

2019年08月19日 17:10:33 Nani_xiao 阅读数 1369更多
分类专栏： 图像处理 深度学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/xiao_lxl/article/details/95970471
文章目录
验证码生成
训练验证码识别器
验证码识别

整理下以前的代码，做个笔记：
验证码生成
利用captcha库生成验证码

from captcha.image import ImageCaptcha #pip install captcha
import numpy as np
import matplotlib.pyplot as plt
from PIL import Image
import random

number = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']
alphabet = ['a', 'b', 'c', 'd', 'e', 'f','g','h','i','j','k','l','m','n','o','p','q','r','s','t','u','v','w','x','y','z']
ALPHABET = ['A','B','C','D','E','F','G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U','V','W','X','Y','Z']

def random_captcha_text(char_set = number+alphabet+ALPHABET, captcha_size=4):
    captcha_text = []
    for i in range(captcha_size):
        c = random.choice(char_set)
        captcha_text.append(c)
    return captcha_text
    
def gen_captcha_text_and_image():
    image = ImageCaptcha()
    
    captcha_text = random_captcha_text()
    captcha_text = ''.join(captcha_text)
    
    captcha = image.generate(captcha_text)
    #image.write(captcha_text, captcha_text + '.jpg')module 'matplotlib' has no attribute 'figure'
    
    captcha_image = Image.open(captcha)
    captcha_image = np.array(captcha_image)
    return captcha_text, captcha_image
    
    
if __name__ == '__main__':
    text, image = gen_captcha_text_and_image()
    
    f = plt.figure()
    ax = f.add_subplot(111)
    ax.text(0.1,0.9,text, ha = 'center', va = 'center', transform=ax.transAxes)
    plt.imshow(image)
    
    plt.show()