2016-03-08 28 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

谢谢,是的,这有效。 –