首页 > 其他分享 >关于tensorflow2.x保存模型及加载模型的方法及对比

关于tensorflow2.x保存模型及加载模型的方法及对比

时间:2023-07-10 21:46:07浏览次数:34  
标签:tensorflow2 load 模型 保存 model 方法 加载

以下方法都是个人实际中测试和使用的方法,tf2版本在2.3~2.7之间

1、model.save() and model.load()

保存模型:这个方法可以直接将训练后的权重和训练的参数保存下来,一般我个人使用的.h5为后缀把模型整个保存下来。

步骤如下:

(1)创建模型,像添加积木一样对模型添加需要的卷积,池化等操作

 (2)配置神经网络的优化器,计算梯度的方法  

 (3)保存模型

 

加载模型:这样保存下来的模型可以直接将h5文件保存下来,不需要先加载模型

 

 

2、model.save_weight() and model.load_weight()

 (1)这里采用继承Model这个类去实现神经网络(比第一种方法更加常用且受规范)

 下面的方法就是当我们保存模型的权重参数,但是没有保存模型的结构

 加载模型

需要先把模型的结构导入过来,再load模型的参数进去才能进行推理

 

3、model.checkpoint

 这个用的比较少,看这样加载模型的方式,可能跟第二种类似

 

 

 

注意:这里的模型都指定了输入的图片的尺寸,如果想输入的图片尺寸不受限制,那么不要使用flatten拉直神经网络,可以使用全卷积来再进行softmax即可

tf.keras.layers.GlobalAveragePooling2D()

标签:tensorflow2,load,模型,保存,model,方法,加载
From: https://www.cnblogs.com/ohj666/p/17534870.html

相关文章

  • ZBrush 3D游戏建模教程:创建女武士模型
    推荐:将NSDT场景编辑器加入你的3D工具链3D工具集:NSDT简石数字孪生介绍 在本教程中,演示创建实时武士角色的流程,该流程基于 KatiSarinS 的原始概念。我将使用各种各样的软件,如ZBrush,Maya,SubstancePainter和Marmoset。在ZBrush,Maya和SubstancePainter中创建风格化的角色参......
  • cesium 加载geoserver wmts
    遇到问题:①加载瓦片服务时,发现{TileRow}/{TileCol}参数不对,刚好是等级大了一倍;解决方法:tileMatrixLabels:["1","2","3","4","5","6","7","8","9","10","11"] ,注意是从1开始,不是从0开始。......
  • 共探AI大模型时代下的挑战与机遇,华为云HCDE与大模型专家面对面
    摘要:近日,华为开发者大会2023(cloud)“开发者生态创新发展圆桌会议”在东莞华为溪流背坡村成功举办。2023年7月8日,华为开发者大会2023(cloud)“开发者生态创新发展圆桌会议”在东莞华为溪流背坡村成功举办。以大模型为代表的的新一轮人工智能技术浪潮汹涌而来,在圆桌会议上,华为技术专......
  • RecyclerView预加载
    列表的内容是由服务器返回的分页数据,每次浏览到当前页的尾部,都会拉取下一页的数据。这中断用户的浏览,不免产生等待。产品希望让这个过程无感知。一种实现方案是预加载,即在一页数据还未看完时就请求下一页数据,让用户感觉列表的内容是无穷的。监听列表滚动状态第一个想到的方案是监......
  • 共探AI大模型时代下的挑战与机遇,华为云HCDE与大模型专家面对面
    摘要:近日,华为开发者大会2023(cloud)“开发者生态创新发展圆桌会议”在东莞华为溪流背坡村成功举办。2023年7月8日,华为开发者大会2023(cloud)“开发者生态创新发展圆桌会议”在东莞华为溪流背坡村成功举办。以大模型为代表的的新一轮人工智能技术浪潮汹涌而来,在圆桌会议上,华为技术专家为......
  • globalmapper加载DOM显示无法确定投影和datum
    Theprojection/datumofthisGeoTIFFfilecouldnotbedeterminedautomatically.Pleaseconfirmtheprojection/datumforthisfile.Checkwithyourdatasupplierforthisinformationifyoudonothaveit.选择loadfromfile,选择arcgis导出的某个矢量文件的.prj文......
  • Cesium学习笔记3——加载topojson和Geojson
    在根目录下新建bucket.css@import"../Build/CesiumUnminified/Widgets/widgets.css";@import"../Build/CesiumUnminified/Widgets/lighter.css";html{height:100%}body{background:#000;color:#eee;font-family:sans-serif;font-size:9pt;padding:0;margin:0;w......
  • 视频直播系统源码,uniapp滚动加载 下拉刷新
    视频直播系统源码,uniapp滚动加载下拉刷新滚动加载滚动加载指的是当用户滑动页面到底部时,自动加载更多数据。在uniapp中,我们可以通过监onReachBottom来实现滚动加载。 onReachBottom页面滚动到底部的事件(不是scroll-view滚到底),常用于下拉下一页数据。onReachBottom使用注意......
  • PMSM永磁同步电机Pi控制,滑模控制simulink模型/无位置传感器模型
    PMSM永磁同步电机Pi控制,滑模控制simulink模型/无位置传感器模型ID:7420646590586026......
  • LCL_Three_Phase_inverter:基于MATLAB/Simulink的LCL三相并网逆变器仿真模型,交流电流内
    LCL_Three_Phase_inverter:基于MATLAB/Simulink的LCL三相并网逆变器仿真模型,交流电流内环才用PR(比例谐振)控制,PWM波采用SVPWM控制,附带对应的参考文献。仿真条件:MATLAB/SimulinkR2015b,购买前如需转成低版本格式请提前告知,谢谢。ID:72100646652125743......