caffegraph
caffegraph copied to clipboard
Load graphical Caffe networks in Torch
caffegraph
Load Caffe networks in Torch7 using nngraph.
After installing Torch, ensure that you have the protobuf
libraries.
On Ubuntu, this might look like
sudo apt-get install libprotobuf-dev protobuf-compiler
or, on Macintosh OperatingSystem 10:
brew install protobuf
You can then install this package using
luarocks install caffegraph
Then, similarly to loadcaffe,
caffegraph = require 'caffegraph'
model = caffegraph.load('deploy_resnet152.prototxt', 'resnet152.caffemodel')
Note that some modules that are loadable using loadcaffe are not yet implemented in caffegraph. You are welcome to submit a PR with any that you feel are missing!
caffe.proto
is used under license from the University of California.