caffe TypeError: unhashable type: 'numpy.ndarray'

Question

当我使用下面 URL 处的代码进行预测时，出现错误：

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模型推断结果时，这种方式比较流行