在Python中不安装caffe的情况下从.caffemodel中提取权重

Question

有没有一种相对简单的方法可以从Caffe Zoo中的许多预训练模型中提取Python中的权重而不使用Caffe（也不使用pyCaffe）？i、 e.将.caffemodel解析为hdf5/numpy或Python可以读取的任何格式？

所有的答案我都用C++代码和咖啡课或PiffCad。 我看过pycaffe的代码，看起来你真的需要caffe来理解二进制代码，这是唯一的解决方案吗？

Answer 1

我刚才不得不解决那个问题。假设你有一个.caffemodel（二进制proto格式），结果很简单。

1. 下载最新的caffe.proto

2. 编译到python库： protoc——python_out=。咖啡原型机

3. 导入和分析

下面的示例代码

import numpy as np
import sys, os
import argparse
import caffe_pb2 as cq

f = open('VGG_ILSVRC_16_layers.caffemodel', 'r')
cq2 = cq.NetParameter()
cq2.ParseFromString(f.read())
f.close()
print "name 1st layer: " + cq2.layers[0].name 

为我制作：

name 1st layer: conv1_1

显然你可以从你的对象中提取任何你想要的东西。我只是打印了第一层的名字作为例子。 另外，您的模型可能是用layer s数组（不推荐使用）或layer数组（没有）来表示层，但是您得到了要点。