在caffe中提取网络权重

Question

我想用提取优化网络的权重python.我有.caffemodel文件,我已经获得net.params,它给了我整个网络的参数.问题是当我为例如第一层调用它时,即net.params['ip2']它给了我:

python

我怎样才能得到权重矩阵而不是指针？

提前致谢,

阿夫欣

Answer 1

您必须使用.prototxt文件和.caffemodel文件读取网络.

net = caffe.Net('path/to/conv.prototxt', 'path/to/conv.caffemodel', caffe.TEST)
W = net.params['con_1'][0].data[...]
b = net.params['con_1'][1].data[...]

有关更多信息,请查看此链接和此链接.