最近在学习 深度学习实现验证码模型训练,因为需要大量的验证码图片,所以要自动生成验证码图片,效果图:
首先安装相应的库包:
#验证码生成库
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()强制刷新缓冲区,立刻进行打印