將tensorflow的ckpt模型存儲為npy的實例
更新時間：2018年07月09日 09:29:05 作者：機器學習的小學生
今天小編就為大家分享一篇將tensorflow的ckpt模型存儲為npy的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
實例如下所示：
#coding=gbk
import numpy as np
import tensorflow as tf
from tensorflow.python import pywrap_tensorflow
checkpoint_path='model.ckpt-5000'#your ckpt path
reader=pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map=reader.get_variable_to_shape_map()
alexnet={}
alexnet_layer = ['conv1','conv2','conv3','conv4','conv5','fc6','fc7','fc8']
add_info = ['weights','biases']
alexnet={'conv1':[[],[]],'conv2':[[],[]],'conv3':[[],[]],'conv4':[[],[]],'conv5':[[],[]],'fc6':[[],[]],'fc7':[[],[]],'fc8':[[],[]]}
for key in var_to_shape_map:
#print ("tensor_name",key)
str_name = key
# 因為模型使用Adam算法優(yōu)化的，在生成的ckpt中，有Adam后綴的tensor
if str_name.find('Adam') > -1:
continue
print('tensor_name:' , str_name)
if str_name.find('/') > -1:
names = str_name.split('/')
# first layer name and weight, bias
layer_name = names[0]
layer_add_info = names[1]
else:
layer_name = str_name
layer_add_info = None
if layer_add_info == 'weights':
alexnet[layer_name][0]=reader.get_tensor(key)
elif layer_add_info == 'biases':
alexnet[layer_name][1] = reader.get_tensor(key)
else:
alexnet[layer_name] = reader.get_tensor(key)
# save npy
np.save('alexnet_pointing04.npy',alexnet)
print('save npy over...')
#print(alexnet['conv1'][0].shape)
#print(alexnet['conv1'][1].shape)
以上這篇將tensorflow的ckpt模型存儲為npy的實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
在 Python 中,我們經(jīng)常會使用函數(shù)的默認參數(shù)來簡化代碼,但你知道嗎默認參數(shù)的行為有時可能會導致一些難以察覺的錯誤,下面小編就為大家介紹一下如何避免這樣的錯誤吧2025-01-01
這篇文章主要介紹了Python實現(xiàn)多線程下載文件的代碼實例,需要的朋友可以參考下2014-06-06
Python是進行數(shù)據(jù)分析的一種很好的語言,主要是因為以數(shù)據(jù)為中心的Python包的奇妙生態(tài)系統(tǒng),本文主要為大家介紹了如何在Pandas中迭代DataFrame中的行,有需要的可以參考下2023-09-09
這篇文章主要介紹了python中heapq堆排算法的實現(xiàn)，該模塊提供了堆排序算法的實現(xiàn)。堆是二叉樹，最大堆中父節(jié)點大于或等于兩個子節(jié)點，最小堆父節(jié)點小于或等于兩個子節(jié)點。下面文章更多詳細介紹，需要的小伙伴可以參考一下2022-05-05
這篇文章主要介紹了python實現(xiàn)合并兩個數(shù)組的方法,實例分析了兩種常用的合并數(shù)組的技巧,非常簡單實用,需要的朋友可以參考下2015-05-05
這篇文章主要介紹了python3在各種服務器環(huán)境中安裝配置過程,源碼包編譯安裝步驟詳解，本文通過圖文并茂的形式給大家介紹的非常詳細，需要的朋友可以參考下2022-01-01
python并發(fā)編程 Process對象的其他屬性方法join方法詳解
這篇文章主要介紹了python并發(fā)編程 Process對象的其他屬性方法join方法詳解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-08-08
策略模式屬于Python-設計模式中行為模式之一，該模式定義了一系列算法，并將每個算法封裝起來，使它們可以相互替換。本文將通過示例詳細講解這一模式，需要的可以參考一下2023-02-02
這篇文章主要介紹了Python格式化輸出字符串方法,結合實例形式總結分析了使用%與format函數(shù)進行字符串格式化操作相關實現(xiàn)技巧與注意事項,需要的朋友可以參考下2018-10-10
遞歸函數(shù)是一種在函數(shù)內(nèi)部調(diào)用自身的編程技巧。在Python中，我們可以使用遞歸函數(shù)來解決一些需要重復執(zhí)行相同操作的問題。遞歸函數(shù)通常包含兩個部分：基本情況和遞歸情況?；厩闆r是指函數(shù)停止調(diào)用自身的條件，而遞歸情況是指函數(shù)調(diào)用自身來解決更小規(guī)模的問題。2023-09-09