TensorFlow函數(shù)教程：tf.io.decode_image

tf.io.decode_image函數(shù)

別名：

• tf.image.decode_image
• tf.io.decode_image

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

定義在：tensorflow/python/ops/image_ops_impl.py。

decode_bmp，decode_gif，decode_jpeg和decode_png的便捷函數(shù)。

檢測圖像是BMP，GIF，JPEG還是PNG，并執(zhí)行適當(dāng)?shù)牟僮鲗⑤斎胱止?jié)string轉(zhuǎn)換為dtype類型的Tensor。

注意：decode_gif返回一個4-D數(shù)組[num_frames, height, width, 3]，而不是decode_bmp，decode_jpeg和decode_png，它們返回3-D數(shù)組[height，width，num_channels]。如果要將GIF文件與BMP，JPEG或PNG文件混合，請確保在構(gòu)建圖形時考慮到這一點(diǎn)。

參數(shù)：

• contents：0-D string。編碼圖像字節(jié)。
• channels：可選的int。默認(rèn)為0。解碼圖像的顏色通道數(shù)。
• dtype：返回的Tensor所需的DType。
• name：操作的名稱（可選）

返回：

dtype類型的Tensor，對于BMP，JPEG和PNG圖像其shape為[height, width, num_channels]，對于GIF圖像，其shape為[num_frames, height, width, 3]。

可能引發(fā)的異常：

• ValueError：不正確的通道數(shù)。