Tensorflow 1.0之后模型文件、權(quán)重?cái)?shù)值的讀取方式
更新時(shí)間：2020年02月12日 08:41:12 作者：地大大劉
今天小編就為大家分享一篇Tensorflow 1.0之后模型文件、權(quán)重?cái)?shù)值的讀取方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
保存的文件有4個(gè)：
checkpoint
model-parameters.bin-46000.data-00000-of-00001
model-parameters.bin-46000.index
model-parameters.bin-46000.meta
讀取代碼：
from tensorflow.python import pywrap_tensorflow
with tf.Graph().as_default():
with tf.Session() as sess:
ckpt = tf.train.get_checkpoint_state('./checkpoint')
if ckpt and ckpt.model_checkpoint_path:
reader = pywrap_tensorflow.NewCheckpointReader('./checkpoint/model-parameters.bin-46000')
all_variables = reader.get_variable_to_shape_map()
w1 = reader.get_tensor("conv1/weight")
print(w1.shape)
print(w1)
print(all_variables)
else:
print('No checkpoint file found')
以上這篇Tensorflow 1.0之后模型文件、權(quán)重?cái)?shù)值的讀取方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
基于spring boot 日志(logback)報(bào)錯(cuò)的解決方式
今天小編就為大家分享一篇基于spring boot 日志(logback)報(bào)錯(cuò)的解決方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-02-02
這篇文章主要介紹了在python中如何建立一個(gè)自己的包，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2023-02-02
Python 編碼規(guī)范(Google Python Style Guide)
本項(xiàng)目并非 Google 官方項(xiàng)目, 而是由國內(nèi)程序員憑熱情創(chuàng)建和維護(hù),對(duì)于想學(xué)習(xí)python的朋友可以參考一下2018-05-05
Python面向?qū)ο缶幊讨嘘P(guān)于類和方法的學(xué)習(xí)筆記
類與類方法是面向?qū)ο蟮木幊陶Z言中必不可少的特性,本文總結(jié)了Python面向?qū)ο缶幊讨嘘P(guān)于類和方法的學(xué)習(xí)筆記,需要的朋友可以參考下2016-06-06
python使用tqdm庫實(shí)現(xiàn)循環(huán)打印進(jìn)度條
tqdm是一個(gè)用于在Python中添加進(jìn)度條的庫，它可以很容易地集成到while循環(huán)中，這篇文章主要介紹了python循環(huán)打印進(jìn)度條,需要的朋友可以參考下2023-05-05
Python實(shí)現(xiàn)登陸文件驗(yàn)證方法
本篇文章中我們給大家分享了關(guān)于Python實(shí)現(xiàn)登陸文件驗(yàn)證的方法和技巧，有興趣的朋友們參考學(xué)習(xí)下。2018-10-10
python使用py2neo查詢Neo4j的節(jié)點(diǎn)、關(guān)系及路徑
本文介紹了使用Py2neo的NodeMatcher和RelationshipMatcher查詢圖中的節(jié)點(diǎn)和關(guān)系，以及通過執(zhí)行Cypher語句的查詢方式。感興趣的小伙伴請(qǐng)看下文2021-08-08
這篇文章主要介紹了利用Python編寫本地音樂播放器，主要以分享完整源碼展開文章內(nèi)容，具有一的參考價(jià)值，需要的小伙伴可以參考一下2022-03-03
這篇文章主要介紹了Python 異常處理的實(shí)例詳解的相關(guān)資料,希望通過本文能幫助到大家掌握Python 異常的知識(shí)，需要的朋友可以參考下2017-09-09
圖文詳解牛頓迭代算法原理及Python實(shí)現(xiàn)
牛頓迭代法又稱為牛頓-拉夫遜（拉弗森）方法，它是牛頓在17世紀(jì)提出的一種在實(shí)數(shù)域和復(fù)數(shù)域上近似求解方程的方法。本文將利用圖文詳解牛頓迭代算法原理及實(shí)現(xiàn)，需要的可以參考一下2022-08-08