Caffe2提取fc7特征

Question

最近，我尝试使用卷积神经网络（CNN）来提取图像的特征。我使用的模型是alexnet，框架是Caffe2 with Python 3.6.4。

问题是我想提取FC7层的特征，它的维度应该是4096。

我该如何解决这个问题？

Answer 1

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

blob = workspace.FetchBlob('fc7')

现在，您可以轻松打印输出形状：

print(blob.shape)