-
Notifications
You must be signed in to change notification settings - Fork 0
/
generateData.py
92 lines (86 loc) · 2.2 KB
/
generateData.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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import cv2
import numpy as np
def generateData(A, counter, dir, className, pointer, images):
# move character vertically
for j in range(-2, 3):
# count += 1
J = cv2.warpAffine(A, np.float32([[1, 0, 0], [0, 1, j]]), A.shape)
cv2.imwrite(
J,
"dataForTrainingbn/"
+ dir
+ "/"
+ className
+ "/"
+ className
+ "_"
+ str(counter)
+ ".png",
)
images[:, :, pointer] = J
pointer += 1
counter += 1
# figure
# cv2.imshow(J)
# move character horizontally
for i in [-2, -1, 1, 2]:
# count += 1
J = cv2.warpAffine(A, np.float32([[1, 0, i], [0, 1, 0]]), A.shape)
cv2.imwrite(
J,
"dataForTrainingbn/"
+ dir
+ "/"
+ className
+ "/"
+ className
+ "_"
+ str(counter)
+ ".png",
)
images[:, :, pointer] = J
pointer += 1
counter += 1
# figure
# cv2.imshow(J)
# move character diagonally
for i in [-2, -1, 1, 2]:
# count += 1
J = cv2.warpAffine(A, np.float32([[1, 0, i], [0, 1, i]]), A.shape)
cv2.imwrite(
J,
"dataForTrainingbn/"
+ dir
+ "/"
+ className
+ "/"
+ className
+ "_"
+ str(counter)
+ ".png",
)
images[:, :, pointer] = J
pointer += 1
counter += 1
# figure
# cv2.imshow(J)
# count += 1
J = cv2.warpAffine(A, np.float32([[1, 0, i], [0, 1, -i]]), A.shape)
cv2.imwrite(
J,
"dataForTrainingbn/"
+ dir
+ "/"
+ className
+ "/"
+ className
+ "_"
+ str(counter)
+ ".png",
)
images[:, :, pointer] = J
pointer += 1
counter += 1
# figure
# cv2.imshow(J)
return counter, pointer, images