Caffe2提取物fc7特征

时间:2018-05-28 08:40:22

标签: python-3.x convolutional-neural-network caffe2

最近,我尝试使用卷积神经网络(CNN)来提取图像的特征。我使用的模型是alexnet,框架是带有Python 3.6.4的Caffe2。

问题是我要提取FC7图层的特征,其尺寸应为4096.

我该如何处理这个问题?

1 个答案:

答案 0 :(得分:1)

您应该使用工作区的FetchBlob(blob_name)功能。在您的情况下(假设FC7的输出名称为'fc7'):

blob = workspace.FetchBlob('fc7')

现在,您可以轻松打印输出形状:

print(blob.shape)