Tensorflow1.0之后模型文件、權重數值的讀取方式是什么

本篇內容主要講解“Tensorflow1.0之后模型文件、權重數值的讀取方式是什么”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實用性強。下面就讓小編來帶大家學習“Tensorflow1.0之后模型文件、權重數值的讀取方式是什么”吧!

創(chuàng)新互聯公司主要從事成都做網站、成都網站設計、網頁設計、企業(yè)做網站、公司建網站等業(yè)務。立足成都服務坡頭,十余年網站建設經驗,價格優(yōu)惠、服務專業(yè),歡迎來電咨詢建站服務:18980820575

保存的文件有4個：

checkpoint

model-parameters.bin-46000.data-00000-of-00001

model-parameters.bin-46000.index

model-parameters.bin-46000.meta

讀取代碼：

from tensorflow.python import pywrap_tensorflowwith 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')

到此，相信大家對“Tensorflow1.0之后模型文件、權重數值的讀取方式是什么”有了更深的了解，不妨來實際操作一番吧！這里是創(chuàng)新互聯網站，更多相關內容可以進入相關頻道進行查詢，關注我們，繼續(xù)學習！

當前名稱：Tensorflow1.0之后模型文件、權重數值的讀取方式是什么
本文網址：http://www.jiaotiyi.com/article/gssgeh.html