-
Notifications
You must be signed in to change notification settings - Fork 8
/
ETL.py
34 lines (28 loc) · 1.21 KB
/
ETL.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
import os
if not os.path.isdir('Data_Images'):
from PIL import Image
exec(open("Lib/Utils.py").read())
os.mkdir('Data_Images')
os.mkdir('Data_Images/Training')
os.mkdir('Data_Images/Training/Anger')
os.mkdir('Data_Images/Training/Disgust')
os.mkdir('Data_Images/Training/Fear')
os.mkdir('Data_Images/Training/Happiness')
os.mkdir('Data_Images/Training/Sadness')
os.mkdir('Data_Images/Training/Surprise')
os.mkdir('Data_Images/Training/Neutral')
path_train = "Data/data_train.csv"
training_imagesRGB, training_labels = etl_data(path_train)
Categories = ['Anger', 'Disgust', 'Fear', 'Happiness', 'Sadness', 'Surprise', 'Neutral']
training_imagesRGB_trunc = np.floor(training_imagesRGB)
for k in range(len(Categories)):
a = training_imagesRGB_trunc[training_labels[:, k] == 1, :, :, :]
for i in range(a.shape[0]):
b = a[i, :, :, :].astype('uint8')
im = Image.fromarray(b, mode='RGB')
im.save("Data_Images/Training/"+Categories[k]+"/Images"+str(i)+".jpeg", subsampling=0, quality=100)
# RESET
for element in dir():
if element[0:2] != "__":
del globals()[element]
del element