如何通过Pycaffe获取图层的输出形状

Question

标题已经包含完整的问题：如何使用Pycaffe获取Caffe模型的给定图层的输出形状？

我有一个caffe.Net()对象，现在我想要模型中特定图层的输出形状。

Answer 1

给定图层名称，您可以通过以下方式获取其索引：

l_idx = list(net._layer_names).index(my_layer_name)

获得l_idx后，您可以获得其输出（又名"top"）：

tops = [(net._blob_names[bi], net.blobs[net._blob_names[bi]].data.shape) 
        for bi in list(net._top_ids(li))]

每个"top"您都可以获得通知

for tn in tops:
  print "output name {} has shape {}".format(tn, net.blobs[tn].data.shape)

可以找到有关如何通过pycaffe接口访问网络结构的更详细示例here。