(原)ubuntu16在torch中使用caffe训练好的模型

转载请注明出处:

http://www.cnblogs.com/darkknightzh/p/5783006.html

之前使用的是torch,由于其他人在caffe上面预训练了inception模型,需要使用caffe的inception模型进行微调。然后网上搜了一下如何将caffe模型load到torch里面。有两种方式(可直接跳转到3查看):

1. https://github.com/szagoruyko/loadcaffe

该网址的不需要安装caffe。应该是根据.prototxt来强制解析.caffemodel文件的吧(瞎猜的。。。)。这个安装比较简单,直接使用luarocks install loadcaffe就行(唯一的依赖就是protobuf:sudo apt-get install libprotobuf-dev protobuf-compiler),但是装完之后,用不了。。。仔细一看,该网址写着:Loading googlenet is supported by。。。应该就是不支持inception模型吧。

代码如下:

require 'loadcaffe'
model = loadcaffe.load('my inception.prototxt', ' my inception.caffemodel', 'nn')
torch.save("net.t7", model)

反正用的时候,提示如下:

(原)ubuntu16在torch中使用caffe训练好的模型

说明:该网址提供了一些已经训练好的模型(下载不一定需要fq,但是从该网址打开对应的连接需要fq),使用训练好的模型是可以成功载入的。同时自动生成对应的lua文件,如my inception.prototxt.lua。使用torch.save后,会生成.t7文件。

2. 该网址给出了如果使用inception模型的网址的链接(并未算作第二种方式):

https://github.com/soumith/inception.torch

但是该程序是直接读已经写好的lua文件,貌似并不能转caffe的inception。

ps:上面这个网址又给出了tensorflow转torch的V3的链接:

https://github.com/Moodstocks/inception-v3.torch

3. https://github.com/szagoruyko/torch-caffe-binding

这个网址给出的也是从.prototxt及.caffemodel来得到torch的模型。但是不会生成lua文件。使用该程序需要安装caffe。

1)由于使用的电脑已经安装了caffe,因而直接运行如下命令安装:

CAFFE_DIR=/*path-to-caffe-root*/ luarocks install caffe

如:

CAFFE_DIR=/usr/local/caffe luarocks install caffe

说明:

a 上面这个直接在终端里面一次性输入就行(开始在.bashrc文件中加入了CAFFE_DIR并重启了电脑。不清楚如果不加入CAFFE_DIR,程序能否运行)

b 此处为caffe的根目录,如下图,不用加上include目录。

(原)ubuntu16在torch中使用caffe训练好的模型

因为CMakeLists.txt里面包含了需要的子文件夹:

(原)ubuntu16在torch中使用caffe训练好的模型

c 中间碰到了一个问题,就是运行上面的命令后,提示如下:

error: #error This file was generated by a newer version of protoc which is
error: #error incompatible with your Protocol Buffer headers. Please update

截图如下:

(原)ubuntu16在torch中使用caffe训练好的模型

https://github.com/BVLC/caffe/issues/19中有人建议需要更新Protocol Buffers,更新了之后,重新运行上面的命令,安装成功了。。。之前安装的版本未2.5.0,重新安装后的为2.6.1。

ps:参见下面网址:

http://www.cnblogs.com/darkknightzh/p/5782992.html

2)直接使用训练好的模型:

require 'caffe'

net = caffe.Net('deploy.prototxt', 'bvlc_alexnet.caffemodel', 'test')
input = torch.FloatTensor(,,,)
output = net:forward(input)
gradOutput = torch.FloatTensor(,,,)
gradInput = net:backward(input, gradOutput)

或者把训练好的模型嵌入到新的网络中:

require 'caffe'

model = nn.Sequential()
model:add(caffe.Net('deploy.prototxt', 'bvlc_alexnet.caffemodel', 'test'))
model:add(nn.Linear(,))

ps:目前能成功载入caffe的inception模型,但是还不清楚能不能使用呢。。。

载入网上下载的模型,不会有Ignoring source layer的提示,如下:

(原)ubuntu16在torch中使用caffe训练好的模型

载入其他人训练的模型,会提示Ignoring source layer XXX,如下:

(原)ubuntu16在torch中使用caffe训练好的模型

更新:刚才和同事说了一下Ignoring的问题,他说是因为里面有些层没有,不用管这个。

pppppps:见了鬼了,刚才碰到了很诡异的一件事:开始用着好好地,结果突然间就出错了,错误如下:

libcaffe.so.1.0.-rc3: cannot open shared object file:No such file or directory

截图如下:

(原)ubuntu16在torch中使用caffe训练好的模型

我XXX,真的莫名其妙啊。即便只使用require "caffe",也还是报上面的错误。当时去找这个库,印象中在caffe安装目录/usr/local/caffe/build/lib里面找过,没找到,中午睡了个觉,刚才一试,程序又可以了。。。我XXX。又去了这个目录,见鬼了,这几个文件出现了:

(原)ubuntu16在torch中使用caffe训练好的模型

我都准备用http://www.cnblogs.com/smartvessel/archive/2011/01/21/1940868.html里面说的方法,去重新链接了。但是重新链接也需要有那个文件才行啊。结果出现这些文件之后,直接运行程序,又可以了。。。哎,不知道是否是硬盘有问题了。

更新:刚才和同事说了一下找不到libcaffe.so,他说,他经常碰到。。。经常碰到。。。经常碰到。。。他说过一会就好了。好吧。

更新1608181641:刚才又出现了require出错。然后赶快去/usr/local/caffe/build/里面看了一下,果然,已经没有lib目录了。。。如下图:

(原)ubuntu16在torch中使用caffe训练好的模型

真不清楚是硬盘的问题还是什么原因。。。

上一篇:关于SilverLight


下一篇:R与数据分析旧笔记(十二)分类 (支持向量机)