mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

caffe wgex fix

Browse Source
This commit is contained in:
Francesco Gatti
2017-07-18 08:45:03 +02:00
parent e83baae1a7
commit 0887199880
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/caffe_weights_exporter.py
+7 -1
View File
@@ -18,6 +18,8 @@ if __name__ == '__main__':
args = parser.parse_args() args = parser.parse_args()
if not os.path.exists(args.output):
os.makedirs(args.output)
print "\n\n ====== NET LOADED ====== " print "\n\n ====== NET LOADED ====== "
net = caffe.Net(args.model, args.weights, caffe.TEST) net = caffe.Net(args.model, args.weights, caffe.TEST)
@@ -26,8 +28,12 @@ if __name__ == '__main__':
for i in xrange(n_lay): for i in xrange(n_lay):
key = net.params.keys()[i] key = net.params.keys()[i]
print "Layer", key print "Layer", key
print " type: ", net.layer_dict[key].type t = net.layer_dict[key].type
print " type: ", t
w = net.params[key][0].data w = net.params[key][0].data
b = net.params[key][1].data b = net.params[key][1].data
print " weights shape:", np.shape(w) print " weights shape:", np.shape(w)
print " bias shape:", np.shape(b) print " bias shape:", np.shape(b)
w.tofile(args.output + "/" + t + str(i) + ".bin", format="f")
b.tofile(args.output + "/" + t + str(i) + ".bias.bin", format="f")