使用我自己的数据进行Tensorflow错误

Question

我一直在使用Tensorflow库来完成这些教程。现在我想使用自己的数据，但是我失败了。这也许是一个菜鸟问题，但我无法弄清楚。

我正在使用此示例：[https]( https://github.com/aymericdamien/TensorFlow-

Examples/blob/master/examples/3%20-%20Neural%20Networks/convolutional_network.py)

//github.com/aymericdamien/TensorFlow-
Examples/blob/master/examples/3%20-%20Neural%20Networks/convolutional_network.py

我想使用自己的图像，以便将我的图像转换为与tensorflow一起使用，我正在使用以下图像：https
:
//github.com/HamedMP/ImageFlow/blob/master/ImageFlow.py

现在，我从此更改示例中的参数：

 n_input = 784
 n_classes = 10

对此：

 n_input = 9216
 n_classes = 2

我这样做是因为我的图像是96 * 96，并且我的图像只有2类

我还会更改权重和偏向所需的数字。

我这样读取数据：

batch_xs = imgReader.read_images(pathname);

imgReader是ImageFlow文件

但是当我尝试运行它时，我给了我一个错误：

 ValueError: Cannot feed value of shape (104, 96, 96, 1) for Tensor
 u'Placeholder:0', which has shape (Dimension(None), Dimension(9216))

我觉得我正在忽略一些小东西，但看不到。

Answer 1

出现此错误的原因是，您尝试提供的数据形状（104 x 96 x 96 x 1）与输入占位符的形状（batch_sizex
9216，其中batch_size可能是可变的）不匹配。

要使其工作，请在运行训练步骤之前添加以下行：

batch_xs = np.reshape(batch_xs, (-1, 9216))

这将使用numpy将读入的图像（batch_sizexhxwx通道的4-D数组）整形batch_size为占位符期望的x 9216元素矩阵。