caffe - 如何将 .caffemodel 导出到其他应用程序？

Question

是否可以翻译 .caffemodel 文件中的信息，以便（例如）Matlab 读取它。也就是说，有没有办法使用 prototxt 的其他东西编写模型并导入使用 Caffe 训练的权重？

如果答案是“不，它是一个二进制文件并且将始终保持这种状态”，是否有一些关于文件结构的文档，以便可以以某种方式提取重要信息？

Answer 1

如您所知，.caffemodel由权重和偏差组成。在给定 prototxt 的情况下，读取 caffemodel 的权重和偏差的一种简单方法是在 Python 中加载网络并读取权重。

您可以使用：

import caffe
net = caffe.Net(<prototxt-file>,<model-file>,<phase>);

并从net.params

资源

Answer 2

我以VGG为例

from caffe.proto import caffe_pb2
net = caffe_pb2.NetParameter()
caffemodel = sys.argv[1]
with open(caffemodel, 'rb') as f:
    net.ParseFromString(f.read())
for i in net.layer:
    print i.ListFields()[0][-1]
#conv1
#relu1
#norm1
#pool1
#conv2
#relu2
#norm2
#pool2
#conv3
#relu3
#conv4
#relu4
#conv5
#relu5
#pool5
#fc6
#relu6
#drop6
#fc7
#relu7
#drop7
#fc8
#prob