获取 pycaffe 层名称或 blob 类型

Question

我有一些名为 net 的 CNN，我想知道 blob 的类型或层的名称。

例如，我可以轻松访问 blob 名称和后续 blob：

for blob in net.blobs:
    print(blob)
    net.blobs[blob]...

或者，我可以访问图层类型：

for x in range(len(net.layers)):
    print(net.layers[x].type)

是否有办法访问这些信息，例如：

net.blobs[blob].type

或者，

net.layers[x].name

谢谢

Answer 1

Blob 没有type。它是一个 blob：N 维数据的容器。你可以寻找net.blobs[blob].data.shape 的shape，或者寻找net.blobs[blob].diff 计算梯度（如果你反向传播梯度......）

层的名称存储在net._layer_names。您可以通过idx = list(net._layer_names).index('my_layer')获取图层的索引。

请参阅this thread 了解更多信息。