diff options
Diffstat (limited to 'scripts')
-rwxr-xr-x | scripts/caffe_data_extractor.py | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/scripts/caffe_data_extractor.py b/scripts/caffe_data_extractor.py new file mode 100755 index 0000000000..09ea0b86b0 --- /dev/null +++ b/scripts/caffe_data_extractor.py @@ -0,0 +1,34 @@ +#!/usr/bin/env python +import argparse + +import caffe +import numpy as np +import scipy.io + + +if __name__ == "__main__": + # Parse arguments + parser = argparse.ArgumentParser('Extract CNN hyper-parameters') + parser.add_argument('-m', dest='modelFile', type=str, required=True, help='Caffe model file') + parser.add_argument('-n', dest='netFile', type=str, required=True, help='Caffe netlist') + args = parser.parse_args() + + # Create Caffe Net + net = caffe.Net(args.netFile, 1, weights=args.modelFile) + + # Read and dump blobs + for name, blobs in net.params.iteritems(): + print 'Name: {0}, Blobs: {1}'.format(name, len(blobs)) + for i in range(len(blobs)): + # Weights + if i == 0: + outname = name + "_w" + # Bias + elif i == 1: + outname = name + "_b" + else: + pass + + print("%s : %s" % (outname, blobs[i].data.shape)) + # Dump as binary + blobs[i].data.tofile(outname + ".dat") |