tensorflow 18 验证码生成

from captcha.image import ImageCaptcha
import numpy as np
from PIL import Image
import random
import sys
import os

number = [str(i) for i in range(10)]

# 随机生成4个字符
def random_captcha_text(char_set=number,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(path):
    image = ImageCaptcha()
    captcha_text = random_captcha_text()
    captcha_text = ''.join(captcha_text)
    captcha = image.generate(captcha_text)
    image.write(captcha_text, 'captcha/images/' + str(captcha_text) + '.jpg')

num = 10000

if not os.path.exists(path):
    os.makedirs(path)
for i in range(num):
    gen_captcha_text_and_image(path)
    sys.stdout.write('\r>> Creating image %d/%d' % (i+1,num))
    sys.stdout.flush()
sys.stdout.write('\n')
sys.stdout.flush()

生成的验证码：