运行在Digits上训练的模型时出现Caffe Web演示错误

时间:2015-07-24 17:00:37

标签: neural-network deep-learning caffe nvidia-digits

我在Digits上训练了一个神经网络模型,它似乎在那里运行良好 然后我导出训练好的模型文件并将它们复制到运行标准caffe网络演示的不同系统中。 我希望能够插入这些文件并让它们在Caffe中运行但是我收到了错误。

具体而言,我将模型复制到bvlc_reference_caffenet.caffemodel,将deploy.prototxt复制到deploy.prototxt,将mean.binaryproto复制到ilsvrc_2012_mean.npy文件中。 但是,当我尝试运行它时,似乎不喜欢error.binaryproto文件的格式,如错误消息所示:

IOError: Failed to interpret file '/home/vagrant/caffe/python/caffe/imagenet/ilsvrc_2012_mean.npy' as a pickle

我在这里做错了什么?在使用caffe之前,我是否需要以某种方式处理来自Digits的mean.binaryproto文件?

1 个答案:

答案 0 :(得分:2)

您需要将.binaryproto文件转换为numpy文件。 有一个很好的示例here使用caffe.iocaffe.proto