2015-07-24 4 views
2

Я обучил модель нейронной сети на цифрах, и она, казалось, там хорошо работала.
Затем я экспортировал подготовленные файлы моделей и скопировал их в другую систему, на которой запущена стандартная демонстрация caffe. Я надеялся, что просто смогу подключить эти файлы и запустить их в Caffe, но я получаю сообщение об ошибке.Ошибка демонстрации веб-сайта Caffe при работе с моделью, прошедшей обучение по цифрам

В частности, я скопировал свою модель в файл bvlc_reference_caffenet.caffemodel, deploy.prototxt в файл deploy.prototxt и файл mean.binaryproto в файл ilsvrc_2012_mean.npy. Однако, когда я пытаюсь запустить его, это, кажется, не нравится формат файла mean.binaryproto, как указано в сообщении об ошибке:

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

что я делаю неправильно здесь? Нужно ли мне обрабатывать файл mean.binaryproto с Digits каким-то образом, прежде чем использовать его с помощью caffe?

ответ

2

Вам необходимо преобразовать файл .binaryproto в файл numpy. Приятный пример here с использованием caffe.io и caffe.proto.