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]])