2016-07-13 13 views
0

我試圖按照Cifar10 example管道。但是,我想用Numpy數組替換文件讀取。有做那幾個好處:我怎麼能養活numpy的陣列預取和緩衝的TensorFlow

  • 簡單的代碼(我想刪除的二進制文件解析)
  • 簡單的圖形和可視化 - >更容易到其他觀衆
  • 小PERF的改善講解(由於I/O和解析)?

這將是一個簡單的方法來做到這一點?

+0

要清楚,要與預處理隊列中刪除的部分,用含CIFAR圖像numpy的陣列更換輸入數據? –

+0

我想保留預處理部分,例如圖像失真。 – Fabian

回答

0

您需要通過既可以得到伸張reshape_image

  • 給它起名字
  • 找到它的默認名稱,與Tensorboard例如

reshaped_image = tf.cast(read_input.uint8image, tf.float32, name="float_image") 

然後你可以餵你的numpy數組使用類似如下的feed_dict:

reshaped_image = tf.get_default_graph().get_tensor_by_name("float_image") 
sess.run(loss, feed_dict={reshaped_image: your_numpy}) 

這同樣適用於標籤。