2016-12-27 13 views
1

VGGで変更されたネットワーク(ここでは3つの浮動小数点数のベクトル)の出力が得られますが、すべてがOKですが、他のレイヤの出力を取得したいとき(最後の畳み込みレイヤconv5_4)の場合、出力ベクトルはすべてゼロになります。 これは、出力を得るためにPythonのコードです:caffe layer zero output-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]) 

私はLENETにVGG-変更からモデルを変えてくれ

答えて

0

を助けてください。問題は解決した。上記の問題を引き起こすのはvggに変更されたようです。