使用TensorFlow将RGBA PNG转换为RGB

Question

我将tensorflow.image模块与TensorFlow v2.0.0一起使用，以使用以下代码打开图像：

import tensorflow as tf
img = tf.image.decode_png(tf.io.read_file("image.png"))

但是此图像为 RGB-A 。如何将其转换为 RGB ？

（最好仅使用TensorFlow）

Answer 1

功能

tf.io.decode_png(
contents, channels=0, dtype=tf.dtypes.uint8, name=None
)

允许您选择要用于导入图像的颜色模型的类型。

属性channels表示解码图像所需的颜色通道数。

可接受的值为：

0：使用PNG编码图像中的通道数。

1：输出灰度图像。

3：输出RGB图像。

4：输出RGBA图像。

如果要将 RGB-A 图片导入为 RGB 图片，只需使用以下代码：

import tensorflow as tf
img = tf.image.decode_png(tf.io.read_file("image.png"), channels=3)