用於獲得乙個pb檔案的所有節點名稱

2021-09-03 09:32:32 字數 1123 閱讀 4967

# -*- coding: utf-8 -*-

"""created on tue dec 18 18:31:13 2018

1、model_dir為模型路徑資料夾,model_name為模型名稱(自定義非如alexnet等訓練實際名稱)

2、寫入到模型路徑下的result.txt檔案內

@author: mr_dogyang

"""import tensorflow as tf

import os

model_dir = 'd:\\tensorflow\\mytensorflow\\mytensorflow\\slim\\satellite'

model_name = 'inception_v3_frozen_graph.pb'

# 讀取並建立乙個圖graph來存放google訓練好的inception_v3模型(函式)

def create_graph():

with tf.gfile.fastgfile(os.path.join(

model_dir, model_name), 'rb') as f:

# 使用tf.graphdef()定義乙個空的graph

graph_def = tf.graphdef()

graph_def.parsefromstring(f.read())

# imports the graph from graph_def into the current default graph.

tf.import_graph_def(graph_def, name='')

# 建立graph

create_graph()

tensor_name_list = [tensor.name for tensor in tf.get_default_graph().as_graph_def().node]

result_file = os.path.join(model_dir, 'result.txt')

with open(result_file, 'w+') as f:

for tensor_name in tensor_name_list:

f.write(tensor_name+'\n')

獲得乙個表的所有欄位名

場景 在除錯較為負責的儲存過程,需要經常select 當表字段較多,但有幾個字段確實不想見到它,以下就是方便了寫select語句 declare query varchar max declare name varchar 50 declare vacancy cursor local for se...

Activiti獲得下乙個節點

用 color red pvmtransition color 表示箭頭的類。而這些類是從 listlist activityimpl.getoutgoingtransitions 從任務id獲得指向外面的連線線 processdefinitionentity.getactivities task ...

遞迴刪除乙個節點以及該節點下的所有節點示例

有的時候刪除資訊的時候,需要把這條資訊下的所有的都刪除,這個時候就需要遞迴刪除了。下面是我在做部門管理中的刪除部門功能的時候寫的一段遞迴刪除乙個部門以及它的所有子部門的一段 僅供大家參考和自己以後備用。下面是我的一段 的展示 複製 如下 修改一條部門資訊 function del bumen id ...