最近,我尝试使用卷积神经网络(CNN)来提取图像的特征。我使用的模型是alexnet,框架是带有Python 3.6.4的Caffe2。
问题是我要提取FC7图层的特征,其尺寸应为4096.
我该如何处理这个问题?
答案 0 :(得分:1)
您应该使用工作区的FetchBlob(blob_name)
功能。在您的情况下(假设FC7的输出名称为'fc7'):
blob = workspace.FetchBlob('fc7')
现在,您可以轻松打印输出形状:
print(blob.shape)