Captcha generiert Bestätigungscode

In diesem Artikel wird eine einfache Methode zum Generieren von Bestätigungscodes mithilfe des Captcha-Moduls vorgestellt.

Version: Ubuntu 16.04 LTS, Python 3.6.7, Captcha 0.3

Captcha installieren: pip captcha installieren

Der Beispielcode lautet wie folgt:

# -*- coding: utf-8 -*-


import cv2
import numpy as np

from captcha.image import ImageCaptcha


def generate_captcha(text='1'):
    """Generate a digit image."""
    capt = ImageCaptcha(width=224, height=224, font_sizes=[200])
    image = capt.generate_image(text)
    image = np.array(image, dtype=np.uint8)
    return image
    
    
if __name__ == '__main__':
    train_output_dir = './train_images/'
    test_output_dir = './test_images/'
    for i in range(9000):
        label = np.random.randint(0, 10)
        image = generate_captcha(str(label))
        image_name = 'image{}_{}.jpg'.format(i+1, label)
        output_path = train_output_dir + image_name
        cv2.imwrite(output_path, image)
        print ("geneTrainData_"+str(i))
    for i in range(1000):
        label = np.random.randint(0, 10)
        image = generate_captcha(str(label))
        image_name = 'image{}_{}.jpg'.format(i+1, label)
        output_path = test_output_dir + image_name
        cv2.imwrite(output_path, image)
        print ("geneTestData_"+str(i))