-
-
Save drcege/5aa1436d72933e2792c951d3c26d7073 to your computer and use it in GitHub Desktop.
Load binary proto .caffemodel to inspect layer names.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import sys | |
| sys.path.insert(0, 'python/') | |
| import caffe | |
| from caffe.proto import caffe_pb2 | |
| net_param = caffe_pb2.NetParameter() | |
| net_str = open('lenet_iter_5000.caffemodel', 'r').read() | |
| net_param.ParseFromString(net_str) | |
| print net_param.layer[0].name # first layer | |
| print net_param.layer[-1].name # last layer |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment