2017-10-10 19 views
1

我在互聯網上搜索了幾天,似乎無法找到將某個圖像添加到使用初始創建的圖形中的人的示例。請讓我知道,如果我嚴重忽略了一些明顯的東西。要但在上下文中的問題,我已經將單個圖像添加到使用初始版本v3訓練的模型中

1)培養了模型,併產生相關的檢查點文件

model.ckpt-10000.data-00000-of-00001 
model.ckpt-10000.index 
model.ckpt-10000.meta 

2)然後我加載模型

tf.reset_default_graph() 
sess = tf.Session() 
saver = tf.train.import_meta_graph(checkpoint_path + "/model.ckpt-10000.meta", clear_devices=True) 
#<tensorflow.python.training.saver.Saver object at 0x11eea89e8> 
sess.run(saver.restore(sess, checkpoint_path + "/model.ckpt-10000")) 

3)該作品正確,所以我加載默認圖,

graph = tf.get_default_graph() 

這裏是我迷路的地方。從這個例子中可以看出,我們必須通過名稱來標識圖的圖層,以將我們的圖像數據傳遞到 - http://cv-tricks.com/tensorflow-tutorial/training-convolutional-neural-network-for-image-classification/

那麼,這些圖層的名稱是什麼?我想他們就像「DecodeJpeg」和「/ tower1/preditions/logits」,但這些都不如猜測。

謝謝你的幫助。

回答

0

保存/恢復前後操作之間映射的標準方式是將它們添加到collections。在https://www.tensorflow.org/api_guides/python/meta_graph中搜索tf.add_to_collectiontf.get_collection。這些示例保存了training_op和logits,但您也可以保存輸入佔位符。

如果您不能重新保存元圖形def並且它沒有任何集合,那麼查看節點名稱和類型(輸入通常是placeholder ops)可能是您可以做的最好的。

相關問題