我使用keras获取.hdf5的模型文件,如何将其转换为.caffemodel?

时间:2017-08-28 09:44:05

标签: keras caffe hdf5

我用keras训练了一个模型,但我必须采取caffe预测,但我不想重新训练模型,所以我想将.HDF5文件转换为.caffemodel

1 个答案:

答案 0 :(得分:0)

nn_tools支持从/到几个NN框架的转换,包括从keras到caffe。它目前仅支持conv和fc图层,但这是一个良好的开端:请参阅keras_to_caffe.py