Description
A small tool to dump Caffe's *.caffemodel to JSON for inspection (*.prototxt files is not needed). Will not print all the weights to make the output concise.
Example
./caffemodel2json.py CAFFE_ROOT/src/caffe/proto/caffe.proto model_name.caffemodel > dump.json
Output would be something like:
protobuf: calling protoc
protobuf: generated
protobuf: imported
model: caffemodel read in memory. Deserialization will take a few minutes. Take a coffee!
model: deserialized
model: json saved
ALLOK. Quitting
Dependencies
- Python
- protobuf (with Python bindings)
- Caffe's caffe.proto (or from a Caffe's fork, it's important if you are inspecting a model that implements new Caffe layers)
Credits
The tool reuses a function from https://github.com/dpp-name/protobuf-json. Great thanks to Paul Dovbush for making it available.