将chainer模型转换为caffe模型

Question

我有一个由＆＃34; chainer＆＃34;生成的模型，但我需要将其转换为＆＃34; caffe＆＃34;我已经在互联网上搜索过，但是没有任何想法。任何人都可以给出一些建议吗？非常感谢。

Answer 1

Chainer现在使用导出功能将链接器计算图导出为caffe格式

from chainer.exporters import caffe

class Model(chainer.Chain):
   def __init__(self):
       super(Model, self).__init__()
       with self.init_scope():
           self.l1 = L.Convolution2D(None, 1, 1, 1, 0)
           self.b2 = L.BatchNormalization(1)
           self.l3 = L.Linear(None, 1)

def __call__(self, x):
   h = F.relu(self.l1(x))
   h = self.b2(h)
   return self.l3(h)

x = chainer.Variable(np.zeros((1, 10, 10, 10), np.float32))
caffe.export(Model(), [x], None, True, 'test')

您可以转到此Chainer Documentation Link！