Shortcuts

write_jpeg

torchvision.io.write_jpeg(input: Tensor, filename: str, quality: int = 75)[source]

Takes an input tensor in CHW layout and saves it in a JPEG file.

Parameters:
  • input (Tensor[channels, image_height, image_width]) – int8 image tensor of c channels, where c must be 1 or 3.

  • filename (str) – Path to save the image.

  • quality (int) – Quality of the resulting JPEG file, it must be a number between 1 and 100. Default: 75

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources