caffe中如何将hdf5格式转换为caffemodel格式？答案

【问题标题】：How to convert hdf5 format to caffemodel format in caffe?caffe中如何将hdf5格式转换为caffemodel格式？
【发布时间】：2016-10-28 13:56:06
【问题描述】：

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

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

【问题讨论】：

【解决方案1】：

您可以在 solver.prototxt 中设置 caffemodel 文件的首选格式。简单设置

snapshot_format: BINARYPROTO

更多信息请参见caffe.proto。

或者，您可以使用python接口（@987654322@）：

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

【讨论】：

Shai 感谢您的回答，实际上在开始时我不知道有不同的快照格式，所以我以 hdf5 格式保存。我运行了 10 万次迭代，所以我无法再次运行它。
但是我找到了解决这个问题的方法，它对我有用。
@RafayZiaMir 你能详细说明你找到的解决方法吗？你会把它作为答案发布吗？
Shai，我认为这种方式不应该是一个答案，因为我在 10 万次迭代后所做的事情是我有一个 hdf5 格式的模型。然后我更改了求解器文件中的快照格式。然后我加载了经过训练的模型，然后开始对其进行微调，但这次它以 caffemodel 格式保存了模型。所以基本上就是你所说的，但我再次微调了我的网络并以 caffemodel 格式保存。