tflearn mismatch tensor shape

Question

奇怪的错误。我正在使用彩色图像，一切正常。现在为了节省时间和速度，我决定使用只有一个维度的灰色图像（在它之前是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：无法为Tensor'InputData / X：0'提供形状值（128,704,800），其形状为'（？，704,800,1）'

Answer 1

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

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

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