tensorflow模型保存、加载之变量重命名实例
更新时间:2020年4月22日 23:19 点击:1333
话不多说,干就完了。
变量重命名的用处?
简单定义:简单来说就是将模型A中的参数parameter_A赋给模型B中的parameter_B
使用场景:当需要使用已经训练好的模型参数,尤其是使用别人训练好的模型参数时,往往别人模型中的参数命名方式与自己当前的命名方式不同,所以在加载模型参数时需要对参数进行重命名,使得代码更简洁易懂。
实现方法:
1)、模型保存
import os import tensorflow as tf weights = tf.Variable(initial_value=tf.truncated_normal(shape=[1024, 2], mean=0.0, stddev=0.1), dtype=tf.float32, name="weights") biases = tf.Variable(initial_value=tf.zeros(shape=[2]), dtype=tf.float32, name="biases") weights_2 = tf.Variable(initial_value=weights.initialized_value(), dtype=tf.float32, name="weights_2") # saver checkpoint if os.path.exists("checkpoints") is False: os.makedirs("checkpoints") saver = tf.train.Saver() with tf.Session() as sess: init_op = [tf.global_variables_initializer()] sess.run(init_op) saver.save(sess=sess, save_path="checkpoints/variable.ckpt")
2)、模型加载(变量名称保持不变)
import tensorflow as tf from matplotlib import pyplot as plt import os current_path = os.path.dirname(os.path.abspath(__file__)) def restore_variable(sess): # need not initilize variable, but need to define the same variable like checkpoint weights = tf.Variable(initial_value=tf.truncated_normal(shape=[1024, 2], mean=0.0, stddev=0.1), dtype=tf.float32, name="weights") biases = tf.Variable(initial_value=tf.zeros(shape=[2]), dtype=tf.float32, name="biases") weights_2 = tf.Variable(initial_value=weights.initialized_value(), dtype=tf.float32, name="weights_2") saver = tf.train.Saver() ckpt_path = os.path.join(current_path, "checkpoints", "variable.ckpt") saver.restore(sess=sess, save_path=ckpt_path) weights_val, weights_2_val = sess.run( [ tf.reshape(weights, shape=[2048]), tf.reshape(weights_2, shape=[2048]) ] ) plt.subplot(1, 2, 1) plt.scatter([i for i in range(len(weights_val))], weights_val) plt.subplot(1, 2, 2) plt.scatter([i for i in range(len(weights_2_val))], weights_2_val) plt.show() if __name__ == '__main__': with tf.Session() as sess: restore_variable(sess)
3)、模型加载(变量重命名)
import tensorflow as tf from matplotlib import pyplot as plt import os current_path = os.path.dirname(os.path.abspath(__file__)) def restore_variable_renamed(sess): conv1_w = tf.Variable(initial_value=tf.truncated_normal(shape=[1024, 2], mean=0.0, stddev=0.1), dtype=tf.float32, name="conv1_w") conv1_b = tf.Variable(initial_value=tf.zeros(shape=[2]), dtype=tf.float32, name="conv1_b") conv2_w = tf.Variable(initial_value=conv1_w.initialized_value(), dtype=tf.float32, name="conv2_w") # variable named 'weights' in ckpt assigned to current variable conv1_w # variable named 'biases' in ckpt assigned to current variable conv1_b # variable named 'weights_2' in ckpt assigned to current variable conv2_w saver = tf.train.Saver({ "weights": conv1_w, "biases": conv1_b, "weights_2": conv2_w }) ckpt_path = os.path.join(current_path, "checkpoints", "variable.ckpt") saver.restore(sess=sess, save_path=ckpt_path) conv1_w__val, conv2_w__val = sess.run( [ tf.reshape(conv1_w, shape=[2048]), tf.reshape(conv2_w, shape=[2048]) ] ) plt.subplot(1, 2, 1) plt.scatter([i for i in range(len(conv1_w__val))], conv1_w__val) plt.subplot(1, 2, 2) plt.scatter([i for i in range(len(conv2_w__val))], conv2_w__val) plt.show() if __name__ == '__main__': with tf.Session() as sess: restore_variable_renamed(sess)
总结:
# 之前模型中叫 'weights'的变量赋值给当前的conv1_w变量
# 之前模型中叫 'biases' 的变量赋值给当前的conv1_b变量
# 之前模型中叫 'weights_2'的变量赋值给当前的conv2_w变量
saver = tf.train.Saver({
"weights": conv1_w,
"biases": conv1_b,
"weights_2": conv2_w
})
以上这篇tensorflow模型保存、加载之变量重命名实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持猪先飞。
相关文章
- 下面本文章来给大家介绍在php中成员变量的一些对比了,文章举了四个例子在这例子中分别对不同成员变量进行测试与获取操作,下面一起来看看。 有如下4个代码示例,你认...2016-11-25
- 本文主要介绍了C#变量命名规则小结,文中介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2021-09-09
- 这篇文章主要介绍了R语言删除指定变量或对象的操作方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-05-06
解决vue的router组件component在import时不能使用变量问题
这篇文章主要介绍了解决vue的router组件component在import时不能使用变量问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-07-27- 这篇文章主要介绍了Vue select 绑定动态变量的实例讲解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-10-23
- 在PHP中,大部分变量类型,如字符串,整型,浮点,数组等都是值类型的,而类和对象是引用类型,在使用的时候,需要注意这一点。看到网友在讨论PHP的&符号,要彻底理解它的用法,就有必要讨论一下变量的两种形式。PHP的变量在内存中是这样...2015-10-23
- 下面小编就为大家带来一篇java中String类型变量的赋值问题介绍。小编觉得挺不错的。现在分享给大家,给大家一个参考。...2016-03-28
- 虽然C#编译器为每个类型都设置了默认类型,但作为面向对象的设计原则,我们还是需要对变量进行正确的初始化。实际上这也是C#推荐的做法...2020-06-25
- 这篇文章主要介绍了python 实现循环定义、赋值多个变量的操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-03-09
- 今天小编就为大家分享一篇解决tensorflow训练时内存持续增加并占满的问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-04-22
- 这篇文章主要介绍了详解ES6实现类的私有变量的几种写法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-02-10
- 这篇文章主要介绍了C++变量,常量,数组和字符串,是C++入门学习中的基础知识,需要的朋友可以参考下,希望能够给你带来帮助...2021-10-21
- 这篇文章主要给大家介绍了关于如何在JS中如何使用css变量以及export之javascript关键字的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下...2021-09-02
- 这篇文章主要给大家介绍了关于Nest.js环境变量配置与序列化的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-02-21
- 这篇文章主要给大家介绍了关于C#设置与获取环境变量的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧。...2020-06-25
- 本篇文章小编为大家介绍,基于条件变量的消息队列 说明介绍。需要的朋友参考一下...2020-04-25
- 这篇文章主要介绍了Perl中的特殊内置变量详细介绍,需要的朋友可以参考下...2020-06-29
- 这篇文章主要介绍了node.js 全局变量的具体使用,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-04-15
- 这篇文章主要介绍了Lua判断变量是否为数字、字符串是否可以转换为数字等,本文讲解了Lua 判断是字符还是数字的方法、Lua判断数字的方法、判断可否转换为数字的方法、判断并且准备一个初值的方法,需要的朋友可以参考下...2020-06-30
- 静态变量只存在于函数作用域内,也就是说,静态变量只存活在栈中。一般的函数内变量在函数结束后会释放,比如局部变量,但是静态变量却不会。就是说,下次再调用这个函数的时候,该变量的值会保留下来。只要在变量前加上关键字st...2015-11-08