【发布时间】:2015-08-03 20:44:44
【问题描述】:
我按照filter visualization 和classification 示例中的说明获取fc6(完全连接的第6 层)对来自预训练模型(bvlc 参考模型)的文件夹中的多个不同图像的响应,但对于所有图像我得到相同的向量。
这是我使用的代码:
import caffe
caffe.set_mode_cpu()
net = caffe.Classifier(MODEL_FILE, PRETRAINED,
mean=np.load(caffe_root + 'python/caffe/imagenet/ilsvrc_2012_mean.npy').mean(1).mean(1),
channel_swap=(2,1,0),
raw_scale=255,
image_dims=(256, 256))
filenames = next(os.walk(path))[2]
fc6Respose=[]
for i in range(0,len(filenames)):
input_image = caffe.io.load_image(path+filenames[i])
scores = net.predict([input_image])
feat = net.blobs['fc6'].data[4]
fc6Respose.append(feat)
PS:有没有什么简单的方法可以将这些数据存储在一个文件(如txt或csv)中,以便以后使用,无需使用Python即可读取和打开?
【问题讨论】:
-
您使用的教程的链接已损坏。请问可以修吗?
标签: python deep-learning caffe conv-neural-network