decode_heic¶
- torchvision.io.decode_heic(input: Tensor, mode: ImageReadMode = ImageReadMode.UNCHANGED) Tensor[原始檔]¶
將 HEIC 影像解碼為 3 維 RGB[A] Tensor。
警告
為了啟用 torchvision 的 AVIF 解碼功能，您需要先執行
pip install torchvision-extra-decoders。只需安裝該包，無需更新您的程式碼。此功能僅在 Linux 上支援，且仍處於 BETA 階段。如有任何問題，請告知我們：https://github.com/pytorch/vision/issues/new/choose。注意 torchvision-extra-decoders 根據 LGPL 許可釋出。輸出 Tensor 的值通常是 uint8 型別，範圍為 [0, 255]。如果影像的位深度超過 8 位，則輸出 Tensor 為 uint16 型別，範圍為 [0, 65535]。由於 PyTorch 中對 uint16 的支援有限，我們建議在此函式後呼叫
torchvision.transforms.v2.functional.to_dtype()並設定scale=True，將解碼後的影像轉換為 uint8 或 float Tensor。- 引數:
input (Tensor[1]) – 一個包含 HEIC 影像原始位元組的一維連續 uint8 Tensor。
mode (str 或 ImageReadMode) – 影像轉換的目標模式，例如 “RGB”。預設值為 “UNCHANGED”。有關可用模式，請參見
ImageReadMode。
- 返回:
解碼後的影像 (Tensor[image_channels, image_height, image_width])
