如何将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")