获取pycaffe图层名称或blob类型

Question

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

例如，我可以轻松访问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。