tensorflow從pb文件導出模型圖

運行下面的程序,將graph信心保存在log目錄: #!/usr/bin/env python # -*- coding:utf-8 -*- import tensorflow as tf from tensorflow.python.platform import gfile model = 'model.pb' #請將這裏的pb文件路徑改爲自己的 graph = tf.get_default_
相關文章
相關標籤/搜索