torch-caffe-binding:在Torch7中使用Caffe

时间:2024-06-04 02:41:11
【文件属性】:

文件名称:torch-caffe-binding:在Torch7中使用Caffe

文件大小:6KB

文件格式:ZIP

更新时间:2024-06-04 02:41:11

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)

网友评论