新手试运行了一个 caffe 的 Python 代码,出现这个错误是怎么回事?

2017-04-23 21:08:46 +08:00
 Reign

完全的 caffe 新手,研究了一天的 caffe ,虽然糊里糊涂的, GitHub 上找了个项目: https://github.com/BestiVictory/ILGnet 在网上找了个 Python 代码照猫画虎写成如下:

import numpy as np

import matplotlib.pyplot as plt

caffe_root = '/opt/caffe/'

import sys

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

import caffe

MODEL_FILE = caffe_root + 'ILGnet/deploy.prototxt'

PRETRAINED = caffe_root + 'ILGnet/ILGnet-AVA2.caffemodel'

IMAGE_FILE = caffe_root+'examples/images/cat.jpg'

mean_file=caffe_root + 'ILGnet/AVA2_mean.npy'

caffe.set_mode_cpu()

net = caffe.Classifier(MODEL_FILE, PRETRAINED,

mean=np.load(mean_file).mean(1).mean(1),

channel_swap=(2,1,0),

raw_scale=255,

image_dims=(227, 227))

input_image = caffe.io.load_image(IMAGE_FILE)

plt.imshow(input_image)

prediction = net.predict([input_image])

plt.plot(prediction[0])

plt.show()

print 'predicted class:', prediction[0].argmax()

然后就出现: F0423 12:33:29.282009 172 insert_splits.cpp:35] Unknown bottom blob 'label' (layer 'loss1/loss', bottom index 1)

完全新手,估计代码错的很离谱,大家就别嘲笑了哈,想问一下这个如果 Python 实现的话该怎样写? V 站高手多,希望能得到解答,真诚谢谢~

4804 次点击
所在节点    Caffe
0 条回复

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/356831

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX