如何通过Pycaffe获取图层的输出形状
如何通过Pycaffe获取图层的输出形状
提问于 2017-12-31 08:33:16
标题已经包含了完整的问题:如何使用Pycaffe获得Caffe模型的给定图层的输出形状？
我有一个caffe.Net()对象，现在我想要模型中特定层的输出形状。
回答 1
Stack Overflow用户
回答已采纳
发布于 2017-12-31 15:03:40
给定图层名称，您可以通过以下方式获取其索引：
l_idx = list(net._layer_names).index(my_layer_name)一旦你有了l_idx，你就可以得到它的输出(也就是"top"s)：
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接口访问net结构的更详细的示例可以在here中找到。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/48039041
复制相关文章
相似问题
腾讯云开发者
