encode_jpeg

torchvision.io.encode_jpeg(input: Union[Tensor, List[Tensor]], quality: int = 75) → Union[Tensor, List[Tensor]]

將 RGB 張量編碼為原始編碼的 jpeg 位元組，在 CPU 或 CUDA 上進行。

注意

傳遞 CUDA 張量清單比重複個別呼叫 encode_jpeg 更有效率。對於 CPU 張量，效能相當。

參數:

• input (Tensor[channels, image_height, image_width] 或 List[Tensor[channels, image_height, image_width]]) – (列表) uint8 圖像 tensor，具有 c 個通道，其中 c 必須為 1 或 3

• quality (int) – 產生的 JPEG 檔案品質。必須是介於 1 到 100 之間的數字。預設值：75

回傳:

包含 JPEG 檔案原始位元組的 (列表) 一維 uint8 tensor。

回傳類型:

output (Tensor[1] 或 list[Tensor[1]])