TensorFlow:如何将图像转换为一维张量?

TensorFlow:如何将图像转换为一维张量?,tensorflow,png,mnist,Tensorflow,Png,Mnist,我最近使用mnist数据集建立了一个预测手写整数的模型。我现在想使用我自己的图像。我的图像是28x28像素(与mnist集类似),但当我尝试使用tf.image.decode_png将它们转换为张量时,我得到了一个3D张量[28,28,4]。通过阅读,我相信额外的4是RGB相关的。我如何转换为[28,28],忽略任何色阶(如果这是实际问题,可能我完全遗漏了一些东西) 谢谢大家! 正如你所说,你得到了一个3D张量,因为你的图像有3个RGB通道。您可以使用类似的工具来获得您想要的。嗨,TWillia

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


谢谢大家!

正如你所说,你得到了一个3D张量,因为你的图像有3个RGB通道。您可以使用类似的工具来获得您想要的。

嗨,TWilliamson,欢迎使用SO!你能分享你的观点吗?一个最小的可重复的例子?