-
大小: 2KB文件類型: .zip金幣: 1下載: 0 次發(fā)布日期: 2021-01-09
- 語(yǔ)言: 其他
- 標(biāo)簽: tensorflow??pb??
資源簡(jiǎn)介
將tensorflow的PNet,RNet,ONet的模型文件:*.data-00000-of-00001,*.index 和 *.meta這3個(gè)文件固化為pb文件。里面詳細(xì)介紹了節(jié)點(diǎn)名稱:output_node_names如何獲取。直接拿來(lái)就可以運(yùn)行。可以直接看看自己模型文件的所有節(jié)點(diǎn)名稱。代碼里有詳細(xì)說(shuō)明
代碼片段和文件信息
#?-*-coding:?utf-8?-*-
“““
????@Project:?tensorflow_models_nets
????@File???:?convert_pb.py
????@Author?:?panjq
????@E-mail?:?pan_jinquan@163.com
????@Date???:?2018-08-29?17:46:50
????@info???:
????-通過(guò)傳入?CKPT?模型的路徑得到模型的圖和變量數(shù)據(jù)
????-通過(guò)?import_meta_graph?導(dǎo)入模型中的圖
????-通過(guò)?saver.restore?從模型中恢復(fù)圖中各個(gè)變量的數(shù)據(jù)
????-通過(guò)?graph_util.convert_variables_to_constants?將模型持久化
“““
import?numpy?as?np
import?tensorflow?as?tf
from?tensorboard.plugins.beholder.im_util?import?read_image
from?tensorflow.python.framework?import?graph_util
resize_height?=?299??#?指定圖片高度
resize_width?=?299??#?指定圖片寬度
depths?=?3
def?freeze_graph_test(pb_path?image_path):
????‘‘‘
????:param?pb_path:pb文件的路徑
????:param?image_path:測(cè)試圖片的路徑
????:return:
????‘‘‘
????with?tf.Graph().as_default():
????????output_graph
?屬性????????????大小?????日期????時(shí)間???名稱
-----------?---------??----------?-----??----
?????文件????????6796??2019-07-10?15:16??model2pbFile.py
- 上一篇:用DQN來(lái)走迷宮
- 下一篇:H3CSE園區(qū)網(wǎng)筆記
評(píng)論
共有 條評(píng)論