caffe layer zero output-python

Question

我得到VGG改變的網絡的輸出（這里是3個浮點數的向量）每件事都沒問題，但是當我只想得到其他層的輸出（例如最后的卷積層conv5_4）而不是最后一層時，輸出向量全為零。 這是獲取輸出的python代碼：

import sys
import cv2
import numpy as np
sys.path.insert(0, '/home/deep/rahim/caffe-master/python')
import caffe
input_image_file = sys.argv[1]
output_file = sys.argv[2]
model_file = '/home/deep/rahim/PGM/Final/feature/deep/snapshot_iter_355000.caffemodel'
deploy_prototxt = '/home/deep/rahim/PGM/Final/feature/deep/deploy.prototxt'
net = caffe.Net(deploy_prototxt, model_file, caffe.TEST)
layer = 'conv5_4'
transformer = caffe.io.Transformer({'data':       net.blobs['data'].data.shape})
transformer.set_transpose('data', (2,0,1))
transformer.set_raw_scale('data', 255.0)
net.blobs['data'].reshape(1,3,40,40)
img = caffe.io.load_image(input_image_file)
net.blobs['data'].data[...] = transformer.preprocess('data', img)
output = net.forward()
print '%s' %(net.blobs[layer].data[0])

請幫我

Answer 1

我將模型從VGG改為LENET。 問題解決了。 似乎改變了vgg導致了上述問題。