快捷導(dǎo)航

將tensorflow的ckpt模型存儲(chǔ)為npy的實(shí)例

更新時(shí)間：2018年07月09日 09:29:05 作者：機(jī)器學(xué)習(xí)的小學(xué)生

今天小編就為大家分享一篇將tensorflow的ckpt模型存儲(chǔ)為npy的實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧

實(shí)例如下所示：

#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
 # 因?yàn)槟Ｐ褪褂肁dam算法優(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模型存儲(chǔ)為npy的實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Python中如何避免默認(rèn)參數(shù)的陷阱
在 Python 中,我們經(jīng)常會(huì)使用函數(shù)的默認(rèn)參數(shù)來簡(jiǎn)化代碼,但你知道嗎默認(rèn)參數(shù)的行為有時(shí)可能會(huì)導(dǎo)致一些難以察覺的錯(cuò)誤,下面小編就為大家介紹一下如何避免這樣的錯(cuò)誤吧
2025-01-01
Python實(shí)現(xiàn)多線程下載文件的代碼實(shí)例
這篇文章主要介紹了Python實(shí)現(xiàn)多線程下載文件的代碼實(shí)例,需要的朋友可以參考下
2014-06-06
Pandas中迭代DataFrame行的方法總結(jié)
Python是進(jìn)行數(shù)據(jù)分析的一種很好的語言,主要是因?yàn)橐詳?shù)據(jù)為中心的Python包的奇妙生態(tài)系統(tǒng),本文主要為大家介紹了如何在Pandas中迭代DataFrame中的行,有需要的可以參考下
2023-09-09
python中heapq堆排算法的實(shí)現(xiàn)
這篇文章主要介紹了python中heapq堆排算法的實(shí)現(xiàn)，該模塊提供了堆排序算法的實(shí)現(xiàn)。堆是二叉樹，最大堆中父節(jié)點(diǎn)大于或等于兩個(gè)子節(jié)點(diǎn)，最小堆父節(jié)點(diǎn)小于或等于兩個(gè)子節(jié)點(diǎn)。下面文章更多詳細(xì)介紹，需要的小伙伴可以參考一下
2022-05-05
python實(shí)現(xiàn)合并兩個(gè)數(shù)組的方法
這篇文章主要介紹了python實(shí)現(xiàn)合并兩個(gè)數(shù)組的方法,實(shí)例分析了兩種常用的合并數(shù)組的技巧,非常簡(jiǎn)單實(shí)用,需要的朋友可以參考下
2015-05-05
python3在各種服務(wù)器環(huán)境中安裝配置過程
這篇文章主要介紹了python3在各種服務(wù)器環(huán)境中安裝配置過程,源碼包編譯安裝步驟詳解，本文通過圖文并茂的形式給大家介紹的非常詳細(xì)，需要的朋友可以參考下
2022-01-01
python并發(fā)編程 Process對(duì)象的其他屬性方法join方法詳解
這篇文章主要介紹了python并發(fā)編程 Process對(duì)象的其他屬性方法join方法詳解,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下
2019-08-08
Python設(shè)計(jì)模式中的策略模式詳解
策略模式屬于Python-設(shè)計(jì)模式中行為模式之一，該模式定義了一系列算法，并將每個(gè)算法封裝起來，使它們可以相互替換。本文將通過示例詳細(xì)講解這一模式，需要的可以參考一下
2023-02-02
Python格式化輸出字符串方法小結(jié)【%與format】
這篇文章主要介紹了Python格式化輸出字符串方法,結(jié)合實(shí)例形式總結(jié)分析了使用%與format函數(shù)進(jìn)行字符串格式化操作相關(guān)實(shí)現(xiàn)技巧與注意事項(xiàng),需要的朋友可以參考下
2018-10-10
python遞歸函數(shù)使用詳解
遞歸函數(shù)是一種在函數(shù)內(nèi)部調(diào)用自身的編程技巧。在Python中，我們可以使用遞歸函數(shù)來解決一些需要重復(fù)執(zhí)行相同操作的問題。遞歸函數(shù)通常包含兩個(gè)部分：基本情況和遞歸情況?；厩闆r是指函數(shù)停止調(diào)用自身的條件，而遞歸情況是指函數(shù)調(diào)用自身來解決更小規(guī)模的問題。
2023-09-09