tensorflow保存模型文件分析

el/2024/4/19 22:26:47

tensorflow模型保存函数为:

tf.train.Saver()

例如下列代码:

import tensorflow as tf
​
v1= tf.Variable(tf.random_normal([784, 200], stddev=0.35), name="v1")
v2= tf.Variable(tf.zeros([200]), name="v2")
v3= tf.Variable(tf.zeros([100]), name="v3")
saver = tf.train.Saver()
with tf.Session() as sess:init_op = tf.global_variables_initializer()sess.run(init_op)saver.save(sess,"checkpoint/model.ckpt",global_step=1)

运行后,保存模型保存,得到三个文件,分别为.data,.meta,.index,

model.ckpt.data-00000-of-00001
model.ckpt.index
model.ckpt.meta

模型加载为:

with tf.Session() as sess:saver.restore(sess, "/checkpoint/model.ckpt")

meta file保存了graph结构,包括 GraphDef, SaverDef等,当存在meta file,我们可以不在文件中定义模型,也可以运行,而如果没有meta file,我们需要定义好模型,再加载data file,得到变量值.

index file为一个 string-string table,table的key值为tensor名,value为BundleEntryProto, BundleEntryProto.

data file保存了模型的所有变量的值.


http://www.ngui.cc/el/5179154.html

相关文章

目标检测之YoloV1论文及tensorflow实现

You Only Look Once: Unified, Real-Time Object Detection Abstract 作者提出了一种新的物体检测方法YOLO。YOLO之前的物体检测方法主要是通过region proposal产生大量的可能包含待检测物体的 potential bounding box,再用分类器去判断每个 bounding box里是否包含有物体,…

迁移学习总结(One Shot Learning, Zero Shot Learning)

转自: https://blog.csdn.net/XJTU_NOC_Wei/article/details/77850221 目录 目录迁移学习 综述Source Data 与Target Data 都有标签 Fine-tuningMultitask LearningSource Data 有标签Target Data 没有标签 Domain-adversarial trainingZero-shot Learning Source D…

YoloV1源码解析

准备阶段数据集的准备: ① 索引文件trainval.txt: 存放的是训练图片的名称 ② 图片文件: ③ 标签文件xml:

YoloV2之TensorFlow版源码解析

准备阶段数据集的准备: ① 索引文件trainval.txt: 存放的是训练图片的名称 ② 图片文件: ③ 标签文件xml: 网络的设计(分类训练与检测训练)

yolo1以及yolo2转载

物体检测论文-YOLO系列(转自: http://hellodfan.com/2017/10/11/%E7%89%A9%E4%BD%93%E6%A3%80%E6%B5%8B%E8%AE%BA%E6%96%87-YOLO%E7%B3%BB%E5%88%97/) 本次论文主要分为两个部分:YOLO和YOLO9000。 YOLO是Rgb大神在Object Detection上的新尝试…

从零起步构建自己的快速语义搜索模型

雷锋网按:本文为 AI研习社 编译的技术博客,原标题 The unreasonable effectiveness of Deep Learning Representations,作者为 Emmanuel Ameisen 。 翻译 | 付腾 王飞 汪鹏 校对 | 余杭 整理 | MY 训练电脑以人类的方式去看图…

(转)零次学习(Zero-Shot Learning)入门

零次学习(Zero-Shot Learning)入门 小栗子 转自:https://zhuanlan.zhihu.com/p/34656727?spm5176.9876270.0.0.399ce44aXsg7cN 谢谢作者 事物的本质反映世界构成的规律 408 人赞了该文章 很久没有更文章了,主要是没有找到…

1x1卷积核如何降低参数量

本文介绍1*1的卷积核与googlenet里面的Inception。正式介绍之前,首先回顾卷积网络的基本概念。 1. 卷积核:可以看作对某个局部的加权求和;它是对应局部感知,它的原理是在观察某个物体时我们既不能观察每个像素也不能一次观察整体…

Hadoop-集群网络配置(转)

转自&#xff1a; 虚拟机中Centos设置静态IP地址方法 (一)虚拟机设置 首先要修改一下虚拟机,<选择虚拟网络编辑器> 对照着上面的图一个一个解释吧. 首先要选择NAT模式,这里我选择的是VMnet8,也可以点击添加网络再添加一个; 然后就是下面的两个对勾✔一定要选上(DHCP…

centos中的防火墙问题

centos从7开始默认用的是firewalld&#xff0c;这个是基于iptables的&#xff0c;虽然有iptables的核心&#xff0c;但是iptables的服务是没安装的。所以你只要停止firewalld服务即可&#xff1a; sudo systemctl stop firewalld.service && sudo systemctl disable fi…