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] or List[Tensor[channels, image_height, image_width]]) – (列表形式的) uint8 影像張量，包含 c 個通道，其中 c 必須為 1 或 3

• quality (int) – 生成 JPEG 檔案(s) 的質量。必須是 1 到 100 之間的數字。預設值: 75

返回:

包含 JPEG 檔案原始位元組的 (列表形式的) 一維 uint8 張量。

返回型別:

輸出 (Tensor[1] or list[Tensor[1]])