-
Notifications
You must be signed in to change notification settings - Fork 0
/
dataset.py
77 lines (51 loc) · 1.69 KB
/
dataset.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
# -*- coding: utf-8 -*-
"""
Created on Mon May 15 17:30:56 2017
@author: richa
"""
import cv2
import os
from PIL import Image
import numpy as np
#read the dataset
def getImagesDatasetFromFile():
pass
def getImagesDataset():
print("#################################")
print("Getting data from dataset:")
x = []
y = []
iterator = os.walk('./machine_written/English/Fnt')
iterator.next()
i = 0
for folder_content in iterator:
folder_path = folder_content[0]
y.append((len(folder_content[2])/2)*[i])
i+=1
print(str(i)+" de 62")
half_folder = folder_content[2][(len(folder_content[2])/2):-1]
for image in half_folder:
im = Image.open(folder_path+'/'+image)
x.append(np.array(im.getdata(),dtype='float32'))
im.close()
x = np.array(x)
y = np.reshape(y,np.size(y))
print("#################################")
return x,y
#functions used to resize the dataset
def resizeImage(filename,sizeTup):
im = Image.open(filename)
#_im = np.reshape(np.array(im.getdata(),dtype='float32'),(128,128))
imrs = cv2.resize(im,sizeTup)
imsv = Image.fromarray(imrs.astype(np.uint8),mode="L")
imsv.save(filename,"PNG")
def resizeDir():
iterator = os.walk('./machine_written/English/Fnt')
iterator.next()
i = 0
for folder_content in iterator:
folder_path = folder_content[0]
i+=1
half_folder = folder_content[2][0:(len(folder_content[2]))]
for image in half_folder:
resizeImage(folder_path+'/'+image,(64,64))