-
大小: 2KB文件類型: .zip金幣: 1下載: 0 次發(fā)布日期: 2021-01-09
- 語言: 其他
- 標簽: tensorflow??pb??
資源簡介
將tensorflow的PNet,RNet,ONet的模型文件：*.data-00000-of-00001，*.index 和 *.meta這3個文件固化為pb文件。里面詳細介紹了節(jié)點名稱：output_node_names如何獲取。直接拿來就可以運行。可以直接看看自己模型文件的所有節(jié)點名稱。代碼里有詳細說明
代碼片段和文件信息
#?-*-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???:
????-通過傳入?CKPT?模型的路徑得到模型的圖和變量數(shù)據(jù)
????-通過?import_meta_graph?導入模型中的圖
????-通過?saver.restore?從模型中恢復圖中各個變量的數(shù)據(jù)
????-通過?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:測試圖片的路徑
????:return:
????‘‘‘
????with?tf.Graph（）.as_default（）:
????????output_graph ?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????文件????????6796??2019-07-10?15:16??model2pbFile.py
- 上一篇：用DQN來走迷宮
- 下一篇：H3CSE園區(qū)網(wǎng)筆記
評論
共有 條評論