Keras模型到tensforflow

问题描述 投票:3回答:1

是否可以将keras模型(网络架构和权重的h5文件)转换为张量流模型?还是在tensorflow中有一个等效的函数来模拟keras的保存?

tensorflow model keras h5py
1个回答
0
投票

是的,有可能,因为Keras(因为它使用Tensorflow作为后端)也建立了计算图。您只需要从Keras模型中获得此图。

“ Keras仅使用一个图形和一个会话。您可以访问该会话通过:K.get_session()。与之相关的图形将是:K.get_session()。图。“

(来自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
© www.soinside.com 2019 - 2024. All rights reserved.