import cv2
import numpy as np
import os
if os._exists('D:\yzm') is not True:
os.makedirs('D:\yzm')
w = 120
h = 60
for n in range(int(input('请输入验证码个数:'))):
a = np.zeros((h,w,3),np.uint8)
for i in range(h):
for j in range(w):
for k in range(3):
a[i][j][k] = np.random.randint(0,255)
font = cv2.FONT_HERSHEY_SIMPLEX
rand = []
name = []
for i in range(4):
rand.append(np.random.randint(0,9))
name.append(str(rand[i]))
cv2.putText(a,name[i],(10+i*25,40),font,1,(np.random.randint(0,255),np.random.randint(0,255),np.random.randint(0,255)),4)
cv2.imwrite('D:\yzm\\'+str(name[0])+str(name[1])+str(name[2])+str(name[3])+'.jpg',a)