-
Notifications
You must be signed in to change notification settings - Fork 0
/
ImportImagenesRGB.py
100 lines (81 loc) · 2.7 KB
/
ImportImagenesRGB.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
93
94
95
96
97
98
99
# coding: utf-8
# In[ ]:
import h5py
import numpy as np
def import_imagenes_RGB():
FF = h5py.File('data_F','r')
CTF = FF.get('Frailejon')
fraile = np.array(CTF)
CTNF = FF.get('NoFrailejon')
nofraile = np.array(CTNF)
# Tomamos el numero de ejemplos para cada clase
n = fraile.shape[0]
m = nofraile.shape[0]
# Inicializamos los arreglos donde vamos a guardar los ejemplos de cada clase
fraile2 = []
nofraile2 = []
r = m
for l in range(2):
if(l==1):
r = n
for i in range(0,r,1):
for j in range(0,5,1):
if (j==0):
r1=0
c1=0
r2=70
c2=70
if(l==0):
x = nofraile[i,r1:r2,c1:c2,]
nofraile2.append(x)
if(l==1):
x = fraile[i,r1:r2,c1:c2,]
fraile2.append(x)
if (j==1):
r1=r1+30
r2=r2+30
if(l==0):
x = nofraile[i,r1:r2,c1:c2,]
nofraile2.append(x)
if(l==1):
x = fraile[i,r1:r2,c1:c2,]
fraile2.append(x)
if(j==2):
c1=c1+30
c2=c2+30
if(l==0):
x = nofraile[i,r1:r2,c1:c2,]
nofraile2.append(x)
if(l==1):
x = fraile[i,r1:r2,c1:c2,]
fraile2.append(x)
if(j==3):
r1=0
r2=70
if(l==0):
x = nofraile[i,r1:r2,c1:c2,]
nofraile2.append(x)
if(l==1):
x = fraile[i,r1:r2,c1:c2,]
fraile2.append(x)
if(j==4):
r1=15
c1=15
r2=85
c2=85
if(l==0):
x = nofraile[i,r1:r2,c1:c2,]
nofraile2.append(x)
if(l==1):
x = fraile[i,r1:r2,c1:c2,]
fraile2.append(x)
nofraile2=np.asarray(nofraile2)
fraile2=np.asarray(fraile2)
CT_x2=np.concatenate((fraile2,nofraile2))
# Aplanamos las imagenes
#CT_x_columna = CT_x2.reshape(CT_x2.shape[0], -1).T
# Normalizamos los datos
CT_xn = CT_x2/255.
CT_y = ([1]*np.array(fraile2).shape[0]+[0]*np.array(nofraile2).shape[0])
CT_y = np.vstack( CT_y ).T
return CT_xn, CT_y