python - 在 Pycaffe 中通过索引访问 blob

Question

我可以使用名称访问网络中的 blob。例如:

net.blobs['data'].data.shape  

返回上一层的形状。
有没有办法通过索引访问图层？我尝试了以下方法:

net.blobs[1].data.shape  

但是我得到了一个无信息的 python 异常

KeyError: 1

我需要以上述方式访问它，因为在运行时，我没有要访问的图层的名称。
有谁知道如何做到这一点吗？

Answer 1

net.blobs 是 Ordered Dictionary对象，因此您无法通过数字索引访问其项目，您需要使用字符串键。

您可以通过net.blobs.keys()获取可用 key 的列表，并且可以使用以下方式访问blob:

net.blobs[net.blobs.keys()[1]].data.shape

但是这样做应该小心:caffe 可能会在模型中引入辅助 blob(具体来说，当一个 "top" 作为 "馈送时，分割层)底部” 到几层)。