如何将 hdf5 转换为 tensorflow 检查点

【问题标题】：How to convert hdf5 to tensorflow checkpoint如何将 hdf5 转换为 tensorflow 检查点
【发布时间】：2018-10-04 15:48:51
【问题描述】：

我正在研究 Mask-rcnn (https://github.com/matterport/Mask_RCNN) 并希望将 h5 格式的 Keras 检查点转换为 Tensorflow 检查点。该怎么做？

【问题讨论】：

标签： tensorflow keras object-detection

【解决方案1】：

你可以尝试加载keras模型，然后照常使用tensorflow session保存：

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")

【讨论】：

【解决方案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")

【讨论】：