如何加载 caffe 模型并转换为 numpy 数组? [英] How do I load a caffe model and convert to a numpy array?
问题描述
我有一个 caffemodel 文件,其中包含 ethereon 的 caffe-tensorflow 转换实用程序不支持的层.我想生成我的 caffemodel 的 numpy 表示.
I have a caffemodel file that contains layers that are not supported by ethereon's caffe-tensorflow conversion utility. I would like to generate a numpy representation of my caffemodel.
我的问题是,如何将 caffemodel 文件(我也有 prototxt,如果有用的话)转换为 numpy 文件?
My question is, how do I convert a caffemodel file (I also have the prototxt, if that is useful) to a numpy file?
附加信息:我安装了 python、带有 python 接口的 caffe 等.我显然没有使用 caffe 的经验.
Additional info: I have python, caffe with python interfaces etc installed. I am clearly not experienced with caffe.
推荐答案
这是一个很好的函数,可以将 caffe 网络转换为 Python 字典列表,因此您可以随意挑选和阅读它:
Here's a nice function that converts a caffe net to a python list of dictionaries, so you can pickle it and read it anyway you want:
import caffe
def shai_net_to_py_readable(prototxt_filename, caffemodel_filename):
net = caffe.Net(prototxt_filename, caffemodel_filename, caffe.TEST) # read the net + weights
pynet_ = []
for li in xrange(len(net.layers)): # for each layer in the net
layer = {} # store layer's information
layer['name'] = net._layer_names[li]
# for each input to the layer (aka "bottom") store its name and shape
layer['bottoms'] = [(net._blob_names[bi], net.blobs[net._blob_names[bi]].data.shape)
for bi in list(net._bottom_ids(li))]
# for each output of the layer (aka "top") store its name and shape
layer['tops'] = [(net._blob_names[bi], net.blobs[net._blob_names[bi]].data.shape)
for bi in list(net._top_ids(li))]
layer['type'] = net.layers[li].type # type of the layer
# the internal parameters of the layer. not all layers has weights.
layer['weights'] = [net.layers[li].blobs[bi].data[...]
for bi in xrange(len(net.layers[li].blobs))]
pynet_.append(layer)
return pynet_
这篇关于如何加载 caffe 模型并转换为 numpy 数组?的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!