本篇內(nèi)容主要講解“Tensorflow1.0之后模型文件、權(quán)重?cái)?shù)值的讀取方式是什么”，感興趣的朋友不妨來看看。本文介紹的方法操作簡(jiǎn)單快捷，實(shí)用性強(qiáng)。下面就讓小編來帶大家學(xué)習(xí)“Tensorflow1.0之后模型文件、權(quán)重?cái)?shù)值的讀取方式是什么”吧!

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

保存的文件有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_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')

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