python实现随机验证码图片生成

最近在学习 深度学习实现验证码模型训练，因为需要大量的验证码图片，所以要自动生成验证码图片，效果图：

首先安装相应的库包：

#验证码生成库
pip install captcha
#进度条显示库，显示图片生成进度
pip install tqdm

code：

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

num = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']

#生成验证码的随机数字列表，验证码图片中有4个数字
def random_text(charset=num, size=4):
    caplist = []
    for i in range(size):
        captcha_text = random.choice(charset)
        caplist.append(captcha_text)
    return caplist

#生成验证码
def create_captext():
    caplist = random_text()
    caplist = ''.join(caplist)#将验证码列表转为字符串
    image = ImageCaptcha()
    captcha = image.generate(caplist)#生成图片
    image.write(caplist,'image2/'+caplist+'.jpg' )#保存图片，图片名即为里面的数字

#手动实现进度条   
for i in range(100):
    create_captext()
    sys.stdout.write("\r start %d/%d" %(i+1, 100))
    sys.stdout.flush()
sys.stdout.write("\n")
sys.stdout.flush()

#自动进度条
from tqdm import tqdm
import time
for i in tqdm(range(100)):
    create_captext()
    time.sleep(0.01)

进度条效果：

sys.stdout.write()输出不会自动换行，没有end,可用转义字符自行控制

 /n 换行

 /r  回车到本行首，可刷新输出

sys.stdout.flush()强制刷新缓冲区

当我们需要打印一些字符时，并不是立刻就打印，而是先将需要打印的字符放入缓冲区，在缓冲区刷新时打印，当缓冲区未满，或者程序运行未结束时，可使用sys.stdout.flush()强制刷新缓冲区，立刻进行打印

进度条学习参考