-
大小: 1KB文件類型: .py金幣: 1下載: 1 次發布日期: 2021-06-09
- 語言: Python
- 標簽: tensorflow??pb??inference??
資源簡介
tensorflow訓練網絡模型 ,生成用于模型預測的pb模型文件,輸入圖片,進行前向預測
代碼片段和文件信息
#?-*-?coding:?utf-8?-*-
“““
Created?on?Wed?Oct?24?17:54:28?2018
@author:?DELL
“““
import?tensorflow?as?tf
import?numpy?as?np
lookup=dict()
with?open(“dict.txt“?“r“)?as?f:
????for?line?in?f:
????????split=line.split()
????????lookup[int(split[0])]=split[1]
with?tf.Session()?as?sess:
????with?open(‘classify_image_graph_def.pb‘?‘rb‘)?as?f:
????????graph_def?=?tf.GraphDef()
????????graph_def.ParseFromString(f.read())?
?#???????print(graph_def)
#????????file=open(‘Output_nod.txt‘‘w‘)??
?#???????file.write(str(graph_def));
????????softmax_tensor?=?tf.get_defau
- 上一篇:基于arcgis10的pythonaddin設計
- 下一篇:尋路.py
評論
共有 條評論