forked from 317070/Twitch-plays-LSD-neural-net
-
Notifications
You must be signed in to change notification settings - Fork 0
/
mat2npy.py
executable file
·42 lines (30 loc) · 1.07 KB
/
mat2npy.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
import numpy as np
import scipy.io
data = scipy.io.loadmat("data/imagenet-vgg-verydeep-16.mat")
print data.keys()
idxs = [0,2,5,7,10,12,14,17,19,21,24,26,28,31,33,35]
params = []
for i in idxs:
W = data['layers'][0][i][0][0][0][0][0]
W = np.transpose(W, (3,2,0,1))
b = data['layers'][0][i][0][0][0][0][1][0]
#W = W[:,:,::-1,::-1]
print W.shape, b.shape
params.extend([W,b])
np.save("data/vgg16.npy",params)
np.save("data/mean.npy",data['normalization'][0][0][0])
np.save("data/classes.npy",data["classes"][0][0].tolist()[1][0])
data = scipy.io.loadmat("data/imagenet-vgg-verydeep-19.mat")
print data.keys()
idxs = [0,2,5,7,10,12,14,16,19,21,23,25,28,30,32,34,37,39,41]
params = []
for i in idxs:
W = data['layers'][0][i][0][0][0][0][0]
W = np.transpose(W, (3,2,0,1))
b = data['layers'][0][i][0][0][0][0][1][0]
#W = W[:,:,::-1,::-1]
print W.shape, b.shape
params.extend([W,b])
np.save("data/vgg19.npy",params)
np.save("data/mean-19.npy",data['normalization'][0][0][0])
np.save("data/classes-19.npy",data["classes"][0][0].tolist()[1][0])