是否可以将keras模型(网络架构和权重的h5文件)转换为张量流模型?或者在张量流中是否存在与keras的model.save相同的函数?
答案 0 :(得分:0)
是的,这是可能的,因为Keras(因为它使用Tensorflow作为后端)也建立了计算图。您只需要从Keras模型中获取此图。
“ Keras仅使用一个图形和一个会话。您可以访问该会话 通过:K.get_session()。与之相关的图形将是: K.get_session()。graph。”
(摘自fchollet:https://github.com/keras-team/keras/issues/3223#issuecomment-232745857)
或者您可以将此图形保存为检查点格式(https://www.tensorflow.org/api_docs/python/tf/train/Saver):
import tensorflow as tf
from keras import backend as K
saver = tf.train.Saver()
sess = K.get_session()
retval = saver.save(sess, ckpt_model_name)
顺便说一句,由于tensorflow 13您可以直接使用keras:
from tensorflow.python.keras import models, layers