python-3.x - Caffe2提取fc7特征

Question

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

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

我该如何处理这个问题？

Answer 1

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

blob = workspace.FetchBlob('fc7')

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

print(blob.shape)