python - 理解caffe中的net_surgery

Question

我正在关注 net_surgery.ipynb caffe 示例，解释如何修改已保存的 .caffemodel 的权重。然而，由于我对 python 还很陌生，所以我无法理解一些语法。

有人可以解释一下下面给出的代码示例中以 conv_params = {pr: ... 开头的第 7 行的含义吗？ (示例来自 net_surgery.ipynb - 步骤 8)。特别是 pr: 是什么？它是(键，值)之类结构中的键吗？

# Load the fully convolutional network to transplant the parameters.
net_full_conv = caffe.Net('net_surgery/bvlc_caffenet_full_conv.prototxt', 
                          '../models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel',
                          caffe.TEST)
params_full_conv = ['fc6-conv', 'fc7-conv', 'fc8-conv']
# conv_params = {name: (weights, biases)}
conv_params = {pr: (net_full_conv.params[pr][0].data, net_full_conv.params[pr][1].data) for pr in params_full_conv}

for conv in params_full_conv:
    print '{} weights are {} dimensional and biases are {} dimensional'.format(conv, conv_params[conv][0].shape, conv_params[conv][1].shape)

Answer 1

您正在努力解决的问题:

conv_params = {pr: (net_full_conv.params[pr][0].data, net_full_conv.params[pr][1].data) for pr in params_full_conv}

定义一个字典conv_params，其中键为'fc6-conv'、'fc7-conv'和'fc8-conv' 。
使用 for 语句 (... for pr in ...) 构造字典称为“字典理解”，您可以找到有关此构造的更多信息 here 。