tf29: 使用tensorboard視覺化inception_v4
阿新 • • 發佈:2019-01-05
你要的答案或許都在這裡:小鵬的部落格目錄
直接看程式碼和圖示吧:
# coding=utf-8 import tensorflow as tf import os import tarfile import requests ''' inception_pretrain_model_url = 'http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz' # 下載inception pretrain模型 inception_pretrain_model_dir = "inception_pretrain" if not os.path.exists(inception_pretrain_model_dir): os.makedirs(inception_pretrain_model_dir) filename = inception_pretrain_model_url.split('/')[-1] filepath = os.path.join(inception_pretrain_model_dir, filename) if not os.path.exists(filepath): print("開始下載: ", filename) r = requests.get(inception_pretrain_model_url, stream=True) with open(filepath, 'wb') as f: for chunk in r.iter_content(chunk_size=1024): if chunk: f.write(chunk) print("下載完成, 開始解壓: ", filename) tarfile.open(filepath, 'r:gz').extractall(inception_pretrain_model_dir) # TensorBoard log目錄 log_dir = 'inception_log' if not os.path.exists(log_dir): os.makedirs(log_dir)''' inception_pretrain_model_dir = "inception_pretrain" if not os.path.exists(inception_pretrain_model_dir): os.makedirs(inception_pretrain_model_dir) # TensorBoard log目錄 log_dir = 'inception_log' if not os.path.exists(log_dir): os.makedirs(log_dir) # 載入inception graph inception_graph_def_file = os.path.join(inception_pretrain_model_dir, 'my_inception_v4_freeze.pb') with tf.Session() as sess: with tf.gfile.FastGFile(inception_graph_def_file, 'rb') as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) tf.import_graph_def(graph_def, name='') writer = tf.summary.FileWriter(log_dir, sess.graph) writer.close()
使用tensorboard檢視:
tensorboard --logdir inception_log
圖示: