對tensorflow 的模型保存和調(diào)用實例講解
更新時間：2018年07月28日 08:55:33 作者：luoganttcc
今天小編就為大家分享一篇對tensorflow 的模型保存和調(diào)用實例講解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
我們通常采用tensorflow來訓練，訓練完之后應當保存模型,即保存模型的記憶（權(quán)重和偏置），這樣就可以來進行人臉識別或語音識別了。
1.模型的保存
# 聲明兩個變量
v1 = tf.Variable(tf.random_normal([1, 2]), name="v1")
v2 = tf.Variable(tf.random_normal([2, 3]), name="v2")
init_op = tf.global_variables_initializer() # 初始化全部變量
saver = tf.train.Saver() # 聲明tf.train.Saver類用于保存模型
with tf.Session() as sess:
sess.run(init_op)
print("v1:", sess.run(v1)) # 打印v1、v2的值一會讀取之后對比
print("v2:", sess.run(v2))
#定義保存路徑，一定要是絕對路徑，且用‘/ '分隔父目錄與子目錄
saver_path = saver.save(sess, "C:/Users/Administrator/Desktop/tt/model.ckpt") # 將模型保存到save/model.ckpt文件
print("Model saved in file:", saver_path)
2.模型的讀取
直接讀取模型時，可能會報錯，我是用Spyder編譯的，可以把Spyder關(guān)掉，再重新打開，就可以讀取數(shù)據(jù)了。原因可能是：在模型保存時將變量初始化了。
import tensorflow as tf
# 使用和保存模型代碼中一樣的方式來聲明變量
v1 = tf.Variable(tf.random_normal([1, 2]), name="v1")
v2 = tf.Variable(tf.random_normal([2, 3]), name="v2")
saver = tf.train.Saver() # 聲明tf.train.Saver類用于保存模型
with tf.Session() as sess:
saver.restore(sess, "C:/Users/Administrator/Desktop/tt/model.ckpt") # 即將固化到硬盤中的Session從保存路徑再讀取出來
print("v1:", sess.run(v1)) # 打印v1、v2的值和之前的進行對比
print("v2:", sess.run(v2))
print("Model Restored")
以上這篇對tensorflow 的模型保存和調(diào)用實例講解就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
Pytorch實現(xiàn)將模型的所有參數(shù)的梯度清0
這篇文章主要介紹了Pytorch實現(xiàn)將模型的所有參數(shù)的梯度清0，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06
Python cookbook(數(shù)據(jù)結(jié)構(gòu)與算法)將名稱映射到序列元素中的方法
這篇文章主要介紹了Python cookbook(數(shù)據(jù)結(jié)構(gòu)與算法)將名稱映射到序列元素中的方法,結(jié)合實例形式分析了Python使用collections.namedtuple()進行元組命名相關(guān)操作技巧,需要的朋友可以參考下2018-03-03
這篇文章主要介紹了python?scrapy框架的日志文件問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08
pandas 是基于NumPy 的一種工具，該工具是為了解決數(shù)據(jù)分析任務(wù)而創(chuàng)建的，這篇文章主要介紹了python中pandas常用命令,需要的朋友可以參考下2022-07-07
生成器（Generators）和迭代器（Iterators）是 Python 中用于處理序列數(shù)據(jù)的強大工具，本文主要來和大家介紹一下它們的具體使用，方便大家更好的了解它們，需要的可以學習下2022-03-03
在 Rest Framework 中提供了兩個 router ， 可以幫助我們快速的實現(xiàn)路由的自動生成，本文通過實例代碼給大家介紹drf-router和authenticate認證源碼分析，感興趣的朋友跟隨小編一起看看吧2021-07-07
在本篇文章里小編給大家整理的是一篇關(guān)于python全棧開發(fā)語法的相關(guān)總結(jié)內(nèi)容，有興趣的朋友們可以跟著學習參考下。2020-11-11
基于OpenCV實現(xiàn)小型的圖像數(shù)據(jù)庫檢索功能
下面就使用VLAD表示圖像，實現(xiàn)一個小型的圖像數(shù)據(jù)庫的檢索程序。下面實現(xiàn)需要的功能模塊，分步驟給大家介紹的非常詳細，對OpenCV圖像數(shù)據(jù)庫檢索功能感興趣的朋友跟隨小編一起看看吧2021-12-12
今天小編就為大家分享一篇python 讀取目錄下csv文件并繪制曲線v111的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-07-07
Python實現(xiàn)xml格式轉(zhuǎn)txt格式的示例代碼
VOC 的標注是xml格式的，而YOLO是.txt格式，所以要實現(xiàn)VOC數(shù)據(jù)集轉(zhuǎn)YOLO數(shù)據(jù)集，只能利用代碼實現(xiàn)。所以本文為大家介紹了Python中xml轉(zhuǎn)txt的示例代碼，需要的可以參考一下2022-03-03