Compute Library
 22.05
caffe_data_extractor.py File Reference

Basic script to export weights from Caffe to npy files. More...

Go to the source code of this file.

Namespaces

 caffe_data_extractor
 

Variables

 parser = argparse.ArgumentParser('Extract Caffe net parameters')
 
 dest
 
 type
 
 str
 
 required
 
 True
 
 help
 
 args = parser.parse_args()
 
 net = caffe.Net(args.netFile, 1, weights=args.modelFile)
 
string outname = name + "_w"
 
string varname = outname
 

Detailed Description

Basic script to export weights from Caffe to npy files.

Definition in file caffe_data_extractor.py.