Python OpencV generates a verification code

The basic idea is to use OpencV to take randomly generated characters, randomly generated line segments, and put them into a randomly generated image.

Although there is no complex morphological processing, but so far the effect seems to be good

Tried to generate 1000 images, but ended up with only 998 because there were duplicates and they were overwritten.

The code is as follows:

import cv2
import numpy as np
line_num = 10
pic_num = 1000
path = "./imgs/"
def randcolor() :        
    return (np.random.randint(0.255),np.random.randint(0.255),np.random.randint(0.255))
    
def randchar() :
    return chr(np.random.randint(65.90))
    
def randpos(x_start,x_end,y_start,y_end) :
    return (np.random.randint(x_start,x_end),
            np.random.randint(y_start,y_end))
    
    
img_heigth = 60
img_width = 240
for i in range(pic_num):
    img_name = ""
    Generate a random matrix, randint(low[, high, size, dtype])
    img = np.random.randint(100.200,(img_heigth,img_width, 3), np.uint8)
    # display image
    #cv2.imshow("ranImg",img)
    
    x_pos = 0
    y_pos = 25
    for i in range(4):
        char = randchar()
        img_name += char
        cv2.putText(img,char,
                    (np.random.randint(x_pos,x_pos + 50),np.random.randint(y_pos,y_pos + 35)), 
                    cv2.FONT_HERSHEY_SIMPLEX,
                    1.5,
                    randcolor(),
                    2,
                    cv2.LINE_AA)
        x_pos += 45
    
    #cv2.imshow("res",img)
    
    # Add line segment
    for i in range(line_num):
        img = cv2.line(img,
                       randpos(0,img_width,0,img_heigth),
                       randpos(0,img_width,0,img_heigth),
                        randcolor(),
                        np.random.randint(1.2))
        
    #cv2.imshow("line",img)
    cv2.imwrite(path + img_name + ".jpg",img)
    #cv2.waitKey(0)                  
    #cv2.destroyAllWindows()
Copy the code

Results: