python自动生成图片验证码训练集和测试集

在OCR识别图片验证码的训练时，需要自己创建验证码，或者到各大网站爬取验证码，下述函数是自动生成图片验证码，生成的验证码如图：

def generate_verifycode_img(img_path, img_num, chars_set, chars_num, H, W):
	"""
    自动生成图片验证码
    :param img_path: 保存验证码图片的路径
    :param img_num: 生成验证码的个数
    :param chars_set: 验证码字符集
    :param chars_num: 验证码字符的个数
    :param H: 生成图片的高
    :param W: 生成图片的宽
    :return: 
    """
    if not os.path.exists(img_path):
        os.makedirs(img_path)
    image = ImageCaptcha(width=W, height=H)
    for i in range(img_num):
        label = ''.join(random.sample(chars_set, chars_num))
        image.write(label, os.path.join(img_path, label + '.png'))