caffe - 将chainer模型转换为caffe模型

Question

我有一个由“chainer”生成的模型，但我需要将其转换为“caffe”模型。我在互联网上搜索过，但没有任何想法。任何人都可以提供一些建议吗？非常感谢。

Answer 1

Chainer 现在使用导出函数将 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 文档链接！