Caffe模型为每个图像提供相同的输出

时间:2017-05-22 17:52:59

标签: deep-learning caffe pycaffe

我刚刚使用预定义的prototxt和caffemodel文件在caffe中加载了alexnet。但是,将任何图像发送到模型会返回与fc7​​图层输出相同的值。

以下是代码段

net=caffe.Net('alexnet/train_val.prototxt','alexnet/bvlc_alexnet.caffemodel',caffe.TEST) 

for image in images:
    im = np.array(caffe.io.load_image(image))
    im = caffe.io.resize_image(im, (227, 227))
    im = np.array(im,dtype=np.float32)
    im =255*im;
    im = im[:,:,::-1]
    im -= np.array(((104.00698793,116.66876762,122.67891434)))
    im = im.transpose(2,0,1)
    net.blobs['data'].reshape(1,*im.shape)
    net.blobs['data'].data[...] = im
    net.forward()
    fc = net.blobs['fc7'].data #Always the same value

0 个答案:

没有答案