Python tf.image.yuv_to_rgb用法及代碼示例

將一幅或多幅圖像從 YUV 轉換為 RGB。

用法

tf.image.yuv_to_rgb(
    images
)

輸出與 images 張量形狀相同的張量，包含像素的 RGB 值。僅當圖像中的 Y 值在 [0,1] 中，U 和 V 值在 [-0.5,0.5] 中時，輸出才被明確定義。

根據上麵的說明，如果 YUV 圖像的像素值不在要求的範圍內，則需要縮放它們。下麵給出的示例說明了在將圖像輸入 yuv_to_rgb 之前對每個圖像通道的預處理。

yuv_images = tf.random.uniform(shape=[100, 64, 64, 3], maxval=255)
last_dimension_axis = len(yuv_images.shape) - 1
yuv_tensor_images = tf.truediv(
    tf.subtract(
        yuv_images,
        tf.reduce_min(yuv_images)
    ),
    tf.subtract(
        tf.reduce_max(yuv_images),
        tf.reduce_min(yuv_images)
     )
)
y, u, v = tf.split(yuv_tensor_images, 3, axis=last_dimension_axis)
target_uv_min, target_uv_max = -0.5, 0.5
u = u * (target_uv_max - target_uv_min) + target_uv_min
v = v * (target_uv_max - target_uv_min) + target_uv_min
preprocessed_yuv_images = tf.concat([y, u, v], axis=last_dimension_axis)
rgb_tensor_images = tf.image.yuv_to_rgb(preprocessed_yuv_images)