Shortcuts

read_image

torchvision.io.read_image(path: str, mode: ImageReadMode = ImageReadMode.UNCHANGED) Tensor[source]

Reads a JPEG or PNG image into a 3 dimensional RGB or grayscale Tensor. Optionally converts the image to the desired format. The values of the output tensor are uint8 in [0, 255].

Parameters:
  • path (str) – path of the JPEG or PNG image.

  • mode (ImageReadMode) – the read mode used for optionally converting the image. Default: ImageReadMode.UNCHANGED. See ImageReadMode class for more information on various available modes.

Returns:

output (Tensor[image_channels, image_height, image_width])

Examples using read_image:

Tensor transforms and JIT

Tensor transforms and JIT

Tensor transforms and JIT
Repurposing masks into bounding boxes

Repurposing masks into bounding boxes

Repurposing masks into bounding boxes
Visualization utilities

Visualization utilities

Visualization utilities

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