-
Notifications
You must be signed in to change notification settings - Fork 4
/
generate_data.py
44 lines (32 loc) · 1023 Bytes
/
generate_data.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
# 2020-11-19
# wangping
# generate digital images
import random
import os
from PIL import Image, ImageDraw
random.seed(3)
path_img = "data/"
def mkdir_for_imgs():
for i in range(10):
if os.path.isdir(path_img + str(i)):
pass
else:
os.mkdir(path_img + str(i))
def generate_single():
im_new = Image.new('L', (32,32),128)
draw = ImageDraw.Draw(im_new)
num = str(random.randint(0, 9))
draw.text(xy=(12, 12), text=num)
return im_new, num
def generate_nums(n):
cnt_num = [0] * 10
for m in range(n):
im, generate_num = generate_single()
for j in range(10):
if generate_num == str(j):
cnt_num[j] = cnt_num[j] + 1
print("Generate:", path_img + str(j) + "/" + str(j) + "_" + str(cnt_num[j]) + ".png")
im.save(path_img + str(j) + "/" + str(j) + "_" + str(cnt_num[j]) + ".png")
# generate n times
mkdir_for_imgs()
generate_nums(100)