tensorflow - 如何将图像导入 Tensorflow？

Question

我是 Tensorflow 新手，正在学习 Mnist 简单教程。现在我想用我自己的图像做类似的事情。我不知道该怎么做。本教程是这样做的:

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

当我从图像创建它时，batch_xs、batch_ys 到底应该是什么？

我看到有一个 ImageFlow 库，这似乎完全符合我的要求，但我也不知道如何使用它。描述说我应该打电话

  convert_images(images, labels, filename)

但它甚至不包含我的图像的路径。

非常感谢您的关注。

Answer 1

查看 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_xs将为100 x 784， batch_ys 将为 100 x 10。对于您自己的应用程序，您可能需要更改像素数和类数。