Keras模型到tensforflow

时间:2017-09-30 07:00:48

标签: tensorflow model keras h5py

是否可以将keras模型(网络架构和权重的h5文件)转换为张量流模型?或者在张量流中是否存在与keras的model.save相同的函数?

1 个答案:

答案 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