-
Notifications
You must be signed in to change notification settings - Fork 0
/
Append.py
58 lines (50 loc) · 2.21 KB
/
Append.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
__author__ = 'smithe3'
import cv2
import numpy as np
import imutil
with open("assets/DataSet.txt", "w") as doc:
letters = []
for i in range(1, 63):
for j in range(1,56):
if(i<10):
if(j<10):
address = "assets/English 2/Hnd/Img/Sample00" + str(i) + "/img00" + str(i) + "-00" + str(j) + ".png"
#print address
image = cv2.imread(address, 0)
#print image
cropped = imutil.get_char_simple(image)
resize = cv2.resize(cropped, (20, 20))
reshape = np.reshape(resize, (1, 400))
letters.append(reshape)
else:
address = "assets/English 2/Hnd/Img/Sample00" + str(i) + "/img00" + str(i) + "-0" + str(j) + ".png"
image = cv2.imread(address, 0)
cropped = imutil.get_char_simple(image)
resize = cv2.resize(cropped, (20, 20))
reshape = np.reshape(resize, (1, 400))
letters.append(reshape)
else:
if(j<10):
address = "assets/English 2/Hnd/Img/Sample0" + str(i) + "/img0" + str(i) + "-00" + str(j) + ".png"
image = cv2.imread(address, 0)
cropped = imutil.get_char_simple(image)
resize = cv2.resize(cropped, (20, 20))
reshape = np.reshape(resize, (1, 400))
letters.append(reshape)
else:
address = "assets/English 2/Hnd/Img/Sample0" + str(i) + "/img0" + str(i) + "-0" + str(j) + ".png"
image =cv2.imread(address, 0)
cropped = imutil.get_char_simple(image)
resize = cv2.resize(cropped, (20, 20))
reshape = np.reshape(resize, (1, 400))
letters.append(reshape)
#print j
print "loaded type " + str(i)
for letter in letters:
cropped = str(letter)[2:len(str(letter))-2]
print type(cropped)
doc.write(cropped)
doc.write("\n")
if(len(cropped) != 1643):
print len(cropped)
doc.close()