tensorflow 模型權(quán)重導(dǎo)出實例
更新時間：2020年01月24日 09:34:01 作者：張中星
今天小編就為大家分享一篇tensorflow 模型權(quán)重導(dǎo)出實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
tensorflow在保存權(quán)重模型時多使用tf.train.Saver().save 函數(shù)進行權(quán)重保存，保存的ckpt文件無法直接打開，不利于將模型權(quán)重導(dǎo)入到其他框架使用（如Caffe、Keras等）。
好在tensorflow提供了相關(guān)函數(shù) tf.train.NewCheckpointReader 可以對ckpt文件進行權(quán)重查看，因此可以通過該函數(shù)進行數(shù)據(jù)導(dǎo)出。
import tensorflow as tf
import h5py
cpktLogFileName = r'./checkpoint/checkpoint' #cpkt 文件路徑
with open(cpktLogFileName, 'r') as f:
#權(quán)重節(jié)點往往會保留多個epoch的數(shù)據(jù)，此處獲取最后的權(quán)重數(shù)據(jù)
cpktFileName = f.readline().split('"')[1]
h5FileName = r'./model/net_classification.h5'
reader = tf.train.NewCheckpointReader(cpktFileName)
f = h5py.File(h5FileName, 'w')
t_g = None
for key in sorted(reader.get_variable_to_shape_map()):
# 權(quán)重名稱需根據(jù)自己網(wǎng)絡(luò)名稱自行修改
if key.endswith('w') or key.endswith('biases'):
keySplits = key.split(r'/')
keyDict = keySplits[1] + '/' + keySplits[1] + '/' + keySplits[2]
f[keyDict] = reader.get_tensor(key）
以上這篇tensorflow 模型權(quán)重導(dǎo)出實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
python selenium登錄豆瓣網(wǎng)過程解析
這篇文章主要介紹了python selenium登錄豆瓣網(wǎng)過程解析,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-08-08
下面小編就為大家?guī)硪黄狿ython文件和流(實例講解)。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-09-09
這篇文章主要介紹了Python實現(xiàn)JavaBeans流程，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)吧2023-01-01
python 將numpy維度不同的數(shù)組相加相乘操作
這篇文章主要介紹了python 將numpy維度不同的數(shù)組相加相乘操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-03-03
這篇文章主要介紹了FFrpc python客戶端lib使用解析,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-08-08
keras Lambda自定義層實現(xiàn)數(shù)據(jù)的切片方式,Lambda傳參數(shù)
這篇文章主要介紹了keras Lambda自定義層實現(xiàn)數(shù)據(jù)的切片方式,Lambda傳參數(shù)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06
這篇文章主要為大家詳細介紹了Python計算多幅圖像柵格值的平均值，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-06-06
這篇文章主要介紹了Python用dilb提取照片上人臉的示例，幫助大家更好的利用python處理人像，感興趣的朋友可以了解下2020-10-10
這篇文章主要介紹了python圖像處理之鏡像實現(xiàn)方法,實例分析了鏡像的實現(xiàn)原理與具體操作方法,需要的朋友可以參考下2015-05-05
Django用內(nèi)置方法實現(xiàn)簡單搜索功能的方法
這篇文章主要介紹了Django用內(nèi)置方法實現(xiàn)簡單搜索功能的方法，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-12-12