machine-learning - 如何在caffe中将hdf5格式转换为caffemodel格式？

Question

基本上我使用 caffe 进行深度学习。现在我为我自己的数据集微调了一个网络，并将我的模型保存为扩展的 hdf5 格式.h5。

现在我想使用 matcaffe 对图像进行分类。matcaffe 只理解训练模型的 .caffemodel 格式。
那么有没有办法将hdf5文件转换为caffemodel？

Answer 1

您可以在solver.prototxt. 简单设置

snapshot_format: BINARYPROTO

有关caffe.proto更多信息，请参阅。

---

或者，您可以使用 python 接口（这比 caffe 的 matlab 接口要好得多，恕我直言）：

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