Ошибка веб-демонстрации 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 из цифр, прежде чем использовать его с caffe?
1 ответ
Вам нужно конвертировать .binaryproto файл в файл NumPy. Вот хороший пример использования caffe.io а также caffe.proto,