如何将tf.keras训练模型转换为具有相同预测结果的咖啡

时间:2020-05-20 07:00:52

标签: python tensorflow caffe tf.keras

有人试图将keras模型转移到caffe吗?

我已经使用基于tf.keras(2.2.4-tf)的网络结构对我的keras模型进行了训练,并且需要将此经过训练的.h5模型转换为caffemodel和prototxt。

我已经尝试过keras2caffe项目(github.com/uhfband/keras2caffe),并成功获取了caffemodel和prototxt。但是,喀拉拉邦和咖啡的预测结果完全不同。

还有其他方法可以让我获得正确的转移的caffe模型吗?

我正在使用的环境如下。

Tensorflow 1.15.0

Tensorflow.keras 2.2.4-tf

Caffe 1.0.0

Numpy 1.18.1

cv2 3.4.3

0 个答案:

没有答案