2016-03-08 73 views
2

我是Tensorflow的新手,我遵循Mnist簡單教程。現在我想用我自己的形象做一些類似的事情。我無法弄清楚如何做到這一點。本教程是這樣的:如何讓我的圖像進入Tensorflow?

batch_xs, batch_ys = mnist.train.next_batch(100) 

從我的圖像創建它時,batch_xs,batch_ys應該是什麼?

我看到有一個庫ImageFlow,這似乎正是我想要的,但我無法弄清楚如何使用它。描述說我應該打電話

convert_images(images, labels, filename) 

但這甚至不包含我的圖像的路徑。

非常感謝您的關注。

+0

我應該改進文檔,但對於您的問題,輸入參數中的「圖像」意味着您應該從其他庫中讀取它,例如'cv2'並將這些圖像數組作爲參數傳遞給'convert_images'方法+它們的標籤以將其轉換爲二進制格式。希望這有助於,如果不讓我知道還有什麼不明確的 –

回答

2

綜觀mnist.train.next_batch()實施,看來batch_xs應大小batch_size X num_pixels的矩陣,和batch_ys應當是尺寸爲矩陣batch_size X num_classes(如果one_hot是True)或長度batch_size的向量(除此以外)。

當你調用mnist.train.next_batch(100)batch_xs100 X 784,並且batch_ys100 X 10。對於您自己的應用程序,您可能需要更改像素數量和類數。

+0

謝謝,是的,這有效。 –