tflearn 不匹配张量形状答案

【问题标题】：tflearn mismatch tensor shapetflearn 不匹配张量形状
【发布时间】：2018-02-26 09:19:52
【问题描述】：

奇怪的错误。我使用的是彩色图像，一切正常。现在为了节省时间和加快过程，我决定使用只有一维的灰色图像（之前是 4）。代码与以前非常相似，只是做了一些更改。但现在我收到一个错误，说形状不一样。有什么帮助吗？

X, Y = image_preloader(dataset_file, image_shape=(704,800), mode="file", categoriacal_labels=True, normalize=True, files_extension="jpeg", gray_scale=True)

network = input_data(shape=[None, 704, 800, 1])

ValueError: 无法为形状为“(?, 704, 800, 1)”的张量“InputData/X:0”提供形状 (128, 704, 800) 的值

【问题讨论】：

标签： image preloader tflearn

【解决方案1】：

我知道这有点老了，但希望它可以帮助你或其他人。

您的灰度图像不再具有 RGB 数据/通道的多个维度，因此您必须通过将其更改为 input_data 形状参数来反映这一点：

network = input_data(shape=[None, 704, 800])

【讨论】：