【人工智能】深度学习框架值TF入门-模型保存与加载

简介: 保存完整的模型有很多应用场景,比如在浏览器中使用TensorFlow.js加载运行,比如在移动设备上使用TensorFlow Lite加载运行。

Keras的方式

Keras版本模型保存与加载

函数

保存模型权重:model.save_weights

保存HDF5文件:model.save

保存pb文件:tf.saved_model

tf.saved_modelmodel.save的区别在于,tf.saved_model格式的模型可以直接用来预测,但是tf.saved_model没有保存优化器配置,而model.save保存了优化器配置,所以整体更大。

保存模型权重方法仅仅保存了模型中的权重,而保存模型文件的model.save可以将模型和优化器一起保存,包括权重(weights)、模型配置(architecture)和优化器配置(optimizer configuration)。这样做的好处是,当你恢复模型时,完全不依赖于原来搭建模型的代码。

保存完整的模型有很多应用场景,比如在浏览器中使用TensorFlow.js加载运行,比如在移动设备上使用TensorFlow Lite加载运行。

使用

保存为pd格式

保存:model.save("要保存的目录名称")

读取:model = tf.keras.models.load_model('保存模型的目录名称')

保存为hdf5格式

同上,只是写的不再是目录名称,而是'xxx.h5'

注意:

如果发生报错:model_config = json.loads(model_config.decode('utf-8'))

原因:tensorflow2.1.0支持h5py<3.0.0 而在安装tensorflow会自动安装h5py 3.1.0 。

旧版本的方式(v1.x)

我没安装这个版本的tensorflow,没有经过测试

函数

函数:tf.train.Saver(var_list=none,max_to_keep=5)

  • 保存和加载模型,文件格式:checkpoint文件(扩展名:ckpt)
  • var_list:指定要保存和还原的变量,可以作为一个dict或一个列表传递
  • max_to_keep:指示要保留的最近检查点文件的最大数量。创建新文件时,会删除较旧的文件,如果无或者0,则保留所有检查点文件。默认为5(保留最新的5个检查点文件)

使用方法

旧版本使用办法)(1.x)

saver = tf.train.Saver()
saver.save(Model,"./路径/文件名.ckpt")
saver.restore(Model,"./路径/文件名.ckpt")


相关文章
|
1天前
|
人工智能 数据挖掘 机器人
同样是人工智能 客户在哪儿AI和GPT等大模型有什么不同
客户在哪儿AI生产的是企业全历史行为数据,同时还针对ToB企业,提供基于企业全历史行为数据的数据分析服务。
|
1天前
|
机器学习/深度学习 TensorFlow 语音技术
使用Python实现深度学习模型:语音合成与语音转换
【7月更文挑战第19天】 使用Python实现深度学习模型:语音合成与语音转换
15 1
|
2天前
|
机器学习/深度学习 监控 算法框架/工具
使用Python实现深度学习模型:人脸识别与人脸表情分析
【7月更文挑战第18天】 使用Python实现深度学习模型:人脸识别与人脸表情分析
13 2
|
3天前
|
机器学习/深度学习 编解码 算法框架/工具
使用Python实现深度学习模型:图像超分辨率与去噪
【7月更文挑战第17天】 使用Python实现深度学习模型:图像超分辨率与去噪
14 4
|
4天前
|
机器学习/深度学习 数据可视化 算法框架/工具
使用Python实现深度学习模型:视频处理与动作识别
【7月更文挑战第16天】 使用Python实现深度学习模型:视频处理与动作识别
34 17
|
5天前
|
机器学习/深度学习 人工智能 自然语言处理
算法金 | 秒懂 AI - 深度学习五大模型:RNN、CNN、Transformer、BERT、GPT 简介
**RNN**,1986年提出,用于序列数据,如语言模型和语音识别,但原始模型有梯度消失问题。**LSTM**和**GRU**通过门控解决了此问题。 **CNN**,1989年引入,擅长图像处理,卷积层和池化层提取特征,经典应用包括图像分类和物体检测,如LeNet-5。 **Transformer**,2017年由Google推出,自注意力机制实现并行计算,优化了NLP效率,如机器翻译。 **BERT**,2018年Google的双向预训练模型,通过掩码语言模型改进上下文理解,适用于问答和文本分类。
37 9
|
18小时前
|
机器学习/深度学习 监控 自动驾驶
探索深度学习在图像识别中的应用
【7月更文挑战第20天】本文深入探讨了深度学习技术在图像识别领域的应用,并分析了其背后的关键技术和挑战。通过实例分析,文章揭示了深度学习如何提高图像处理的准确度和效率,并预测了未来技术的发展趋势。
|
18小时前
|
机器学习/深度学习 监控 自动驾驶
探索深度学习在图像识别中的应用
【7月更文挑战第20天】随着人工智能技术的飞速发展,深度学习已成为推动图像识别技术进步的核心动力。本文将深入探讨深度学习如何革新图像识别领域,包括其在处理复杂图像数据、提高识别准确率以及在自动驾驶和医疗诊断等实际应用场景中的运用。文章旨在为读者提供一个关于深度学习在图像识别中应用的全面视角,展示其背后的技术原理与未来发展潜力。
|
1天前
|
机器学习/深度学习 自然语言处理 语音技术
深度学习在自然语言处理中的应用与挑战
本文深入探讨了深度学习技术在自然语言处理(NLP)领域的应用及其所面临的挑战。通过分析深度学习模型如循环神经网络(RNN)、长短时记忆网络(LSTM)和变压器模型(Transformer)等在语言理解、机器翻译、情感分析等方面的应用,揭示了这些技术如何推动NLP的进步。同时,本文也指出了深度学习在处理自然语言时的局限性,包括数据依赖性、解释性不足以及模型泛化能力的问题,并提出了可能的解决方向,旨在为未来NLP的研究提供参考。
|
1天前
|
机器学习/深度学习 自然语言处理 算法
深度学习在自然语言处理中的应用与挑战
本文将深入探讨深度学习技术在自然语言处理(NLP)领域的应用及其面临的挑战。我们将分析深度学习如何革新了语言模型、机器翻译和情感分析等NLP核心任务,并讨论数据依赖性、模型可解释性和计算资源限制等问题。
8 0