caffe TypeError：不可用类型：'numpy.ndarray'

Question

当我使用下面的网址中的代码进行预测时，我收到错误：

https://github.com/BVLC/caffe/wiki/Using-a-Trained-Network:-Deploy

以下是代码：

import caffe
from cv2 import imread

model = 'lenet_train_test.prototxt'
weights = 'lenet_iter_10000.caffemodel'
net  = caffe.Net(model,weights,'test')
image = imread('test1.png')
res = net.forward({image})

这是我得到的错误：

TypeError: unhashable type: 'numpy.ndarray'

当我做{image}

时发生了

Answer 1

他们在互联网中有很多例子。我带了其中一个。

image = np.array(Image.open('sample/image/yours.jpg'))
im_input = im[np.newaxis, np.newaxis, :, :]

net.blobs['data'].reshape(*im_input.shape)
net.blobs['data'].data[...] = im_input

output = net.forward()

这种方式在推理caffe模型的结果时更受欢迎。