caffe - 如何提取网络的权重？

Question

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

<caffe._caffe.BlobVec object at 0x7f1cb03c8fa0>

如何获得权重矩阵而不是指针？

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[...]

看看这个 link 和这个 link 了解更多信息。