【文件属性】:
文件名称:torch-caffe-binding:在Torch7中使用Caffe
文件大小:6KB
文件格式:ZIP
更新时间:2021-05-15 08:54:31
C++
火炬与咖啡的绑定
将Caffe用作Torch7中的模块的简短绑定。 具有与MATLAB绑定相同的功能。
您必须已经安装并构建了Caffe,然后执行以下操作:
CAFFE_DIR=/ * path-to-caffe-root * / luarocks install caffe
支持前进和后退:
require ' caffe '
net = caffe. Net ( ' deploy.prototxt ' , ' bvlc_alexnet.caffemodel ' , ' test ' )
input = torch. FloatTensor ( 10 , 3 , 227 , 227 )
output = net: forward (input)
gradOutput = torch. FloatTensor ( 10 , 1000 , 1 , 1 )
gradInput = n
【文件预览】:
torch-caffe-binding-master
----test.lua(349B)
----ffi.lua(491B)
----LICENSE(1KB)
----caffe.lua(1KB)
----caffe.cpp(5KB)
----CMakeLists.txt(1KB)
----README.md(898B)
----caffe-1.0-0.rockspec(667B)
----init.lua(62B)