【发布时间】:2020-03-18 18:44:57
【问题描述】:
您好,我正在尝试使用 python 将 MNIST 数据集转换为灰度,但我做不到。
代码:
(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train = x_train.reshape(60000, 28, 28, 1)
x_test = x_test.reshape(10000, 28, 28, 1)
当我输入上述内容时,出现此错误:
TypeError:图像数据的形状无效 (28, 28, 1)
我做错了什么?
【问题讨论】:
-
重塑方法不会“聚合”数据。看看这是否有帮助:stackoverflow.com/questions/41971663/…
-
谢谢,但我没有得到答案。我可以在数据集上应用 np.dot() 将其转换为灰度吗?
-
是的,您选择要用于转换的 RGB 比率。虽然我很惊讶你需要这样做,因为 MNIST 数据通常已经是灰度的。也许还要检查您的原始数据维度并提及您从中获取 MNIST 数据的来源。
-
非常感谢您的回复。我使用 keras 加载 MNIST 数据集。
-
可以在此处找到完整的解决方案列表:stackoverflow.com/questions/12201577/…
标签: python deep-learning dataset reshape