Ошибка веб-демонстрации Caffe при запуске модели, обученной цифрам

Я обучил модель нейронной сети на Digits, и она там работала нормально.
Затем я экспортировал файлы обученной модели и скопировал их в другую систему, где была запущена стандартная веб-демонстрация 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?


person user2284821    schedule 24.07.2015    source источник


Ответы (1)


Вам нужно преобразовать файл .binaryproto в файл numpy. Есть хороший пример здесь с использованием caffe.io и caffe.proto.

person Shai    schedule 18.11.2015