如何将hdf5转换为tensorflow检查点

Question

我正在Mask-rcnn (https://github.com/matterport/Mask_RCNN)上工作，希望将h5格式的Keras检查点转换为Tensorflow检查点。如何做到这一点？

Answer 1

您可以尝试加载keras模型，然后像往常一样使用tensorflow会话保存它：

model = keras.models.load_model("mask_rcnn_coco.h5")
session = keras.backend.get_session()

saver = tf.train.Saver()
saver.save(session, "checkpoints/mask_rcnn.ckpt")

Answer 2

这段代码在tensorflow 2.x中对我很有效

import tensorflow as tf
from keras.models import load_model

saver = tf.train.Checkpoint()
model = load_model('model.hdf5', compile=False)
sess = tf.compat.v1.keras.backend.get_session()

save_path = saver.save("model.ckpt")