python - 获取 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 没有类型。它是一个 blob:N 维数据的容器。您可以为它的 shape 寻找 net.blobs[blob].data.shape，或者寻找 net.blobs[blob].diff用于计算梯度(如果你反向传播梯度......)

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

参见 this thread了解更多信息。