Как преобразовать формат hdf5 в формат caffemodel в caffe?

Question

Как преобразовать формат hdf5 в формат caffemodel в caffe?

В основном я использую Caffe для глубокого обучения. Теперь я настроил сеть для своего собственного набора данных и сохранил свою модель в формате hdf5, который .h5 расширение.

Теперь я хочу классифицировать изображения с помощью matcaffe. matcaffe понимает только формат.caffemodel для обученной модели.
Так есть ли способ конвертировать файл hdf5 в caffemodel?

2

machine-learning deep-learning neural-network caffe matcaffe

Источник

user1134354 26 июн '16 в 13:27

1 ответ

Другие вопросы по тегам machine-learning deep-learning neural-network caffe matcaffe

user1714410 27 июн '16 в 06:06 2016-06-27 06:06 · Answer 1 · 2016-06-27 06:06

Вы можете установить предпочтительный формат для файла caffemodel в вашем solver.prototxt, Просто установить

snapshot_format: BINARYPROTO

Увидеть caffe.proto для дополнительной информации.

В качестве альтернативы, вы можете использовать интерфейс Python ( который намного лучше, чем интерфейс Matlab для Caffe, IMHO):

import caffe
net = caffe.Net('/path/to/deploy.prototxt', '/path/to/caffemodel.h5', caffe.TEST)
net.save('/path/to/just.caffemodel')