TensorFlow：如何将图像转换为一维张量？

Question

我最近使用 mnist 数据集构建了一个预测手写整数的模型。我现在想使用我自己的图像。我的图像是 28x28 像素（如 mnist 集），但是当我尝试使用 tf.image.decode_png 将它们转换为张量时，我得到一个 3D 张量 [28, 28, 4]。通过阅读，我相信额外的 4 与 RGB 有关。如何转换为 [28, 28] 忽略任何色标（如果这是实际问题，也许我完全遗漏了一些东西）。

谢谢！

Answer 1

正如你所说的，你得到了一个 3D 张量，因为你的图像有 3 个 RGB 通道。你可以使用tf.image.rgb_to_grayscale 之类的东西来获得你想要的东西。