caffe总是预测相同的结果

时间:2016-02-01 23:20:58

标签: caffe pycaffe

我用我自己的数据训练了我的网络,这不是图像而是矢量(20×128×128)。 现在我想使用训练有素的网络进行预测。 但是,当我尝试代码时, 无论输入矢量是什么,我都得到相同的输出。 我会附上我用过的文件。 任何建议将不胜感激。

import cv2
import numpy as np
import os
import caffe
import lmdb
import leveldb
from caffe.proto import caffe_pb2

caffe_root = '../'
import sys
sys.path.insert(0, caffe_root + 'python')

def read_txt(file_path):
    X = np.empty((20,128,128),dtype=np.float)
    if os.path.exists(file_path) == False:
       return X
    f = open(file_path)
    lines2 = f.readlines()
    f.close()

    count = -1
    for line in lines2:
       count = count + 1
       i = count % 130
       k = count//130
       if i <= 127 and k<20:
           words = line.split(',')
           for j in range(0,128):
           X[k,i,j]=float(words[j])/500000

   return X

MODEL_FILE = '************/try_bn_deploy.prototxt'
PRETRAINED = '*********/bn__iter_20000.caffemodel'
base_path = '*************/database/txt/'

f = open('*****************/val.txt')
line = f.readline()
caffe.set_mode_cpu()
net = caffe.Net(MODEL_FILE, PRETRAINED,caffe.TEST)

count = 0
correct = 0
while line:
  count += 1

  # load feature vector
  pair = line.strip().split()
  image_path = pair[0]
  label = pair[1]
  X=np.empty((20,128,128),dtype=np.float)
  X[:,:,:] = read_txt(base_path + image_path)
  net.blobs['data'].data[...] = X
  # forcast
  out = net.forward()
  plabel = int(out['prob'][0].argmax())
  print count, label, plabel
  line = f.readline()
  f.close`

这是我的部署文件

  name: "Net"
  input: "data"
  input_shape {
     dim: 1
     dim: 20
     dim: 128
     dim: 128
  }
  ...
  layer {
     name: "prob"
     type: "Softmax"
     bottom: "fc8"
     top: "prob"
  }

1 个答案:

答案 0 :(得分:0)

当您致电net.forward()时,它会使用原型文件指定的数据层替换所有图层,从而使您对net.blobs['data'].data[...] = X的调用失效。修补这个运行:

net.forward(data=np.asarray(X))

这应该给你正确的预测。