gpt4 book ai didi

tensorflow - 如何将 hdf5 转换为 tensorflow 检查点

转载 作者:行者123 更新时间:2023-12-04 15:53:58 24 4
gpt4 key购买 nike

我正在研究 Mask-rcnn ( https://github.com/matterport/Mask_RCNN ),并希望将 h5 格式的 Keras 检查点转换为 Tensorflow 检查点。如何做到这一点?

最佳答案

您可以尝试加载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")

关于tensorflow - 如何将 hdf5 转换为 tensorflow 检查点,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52650842/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com