您当前的位置:首页 > IT编程 > Keras
| C语言 | Java | VB | VC | python | Android | TensorFlow | C++ | oracle | 学术与代码 | cnn卷积神经网络 | gnn | 图像修复 | Keras | 数据集 | Neo4j | 自然语言处理 | 深度学习 | 医学CAD | 医学影像 | 超参数 | pointnet | pytorch | 异常检测 | Transformers |

自学教程:h5 to ckpt

51自学网 2023-10-28 23:04:50
  Keras
这篇教程h5 to ckpt写得很实用,希望能帮到您。

3.2.2. h5 to ckpt

  • 参考keras issue9040,fchollet给出了方法。
  • 我自己写的测试样例
import tensorflow as tf
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"   # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"]="1"
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
tf.keras.backend.set_session(tf.Session(config=config))
path_to_save_ckpt = '/path/to/keras-ckpt'

model = tf.keras.applications.VGG16()
model_name = 'vgg16'

var_list = slim.get_variables_to_restore(include=None, exclude=['predictions'])
saver = tf.train.Saver(var_list)
saver.save(tf.keras.backend.get_session(), 
           os.path.join(path_to_save_ckpt, model_name + '.ckpt'))




saver = tf.train.Checkpoint()
model = keras.models.load_model('Model.hdf5', compile=False)
sess = tf.compat.v1.keras.backend.get_session()
save_path = saver.save('model.ckpt')

返回列表
keras学习-不同网络模型融合
51自学网自学EXCEL、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。
京ICP备13026421号-1