快捷键

encode_jpeg

torchvision.io.encode_jpeg(input: Union[Tensor, List[Tensor]], quality: int = 75) Union[Tensor, list[Tensor]][源代码]

将 RGB 张量(s) 编码为原始编码的 jpeg 字节,在 CPU 或 CUDA 上。

注意

传递 CUDA 张量列表比对 encode_jpeg 进行重复的单个调用更有效。对于 CPU 张量,性能相同。

参数::
  • input (Tensor[channels, image_height, image_width] or List[Tensor[channels, image_height, image_width]]) – (列表) uint8 图像张量(s),c 个通道,其中 c 必须为 1 或 3

  • quality (int) – 生成的 JPEG 文件(s) 的质量。必须是 1 到 100 之间的数字。默认值:75

返回::

一个 (列表) 一维 uint8 张量(s),包含 JPEG 文件的原始字节。

返回类型::

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

文档

访问 PyTorch 的综合开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源