Skip to content

nhynes/caffegraph

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

Load graphical Caffe networks in Torch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published