Shortcuts

Utils

The torchvision.utils module contains various utilities, mostly for vizualization.

draw_bounding_boxes(image, boxes[, labels, …])

Draws bounding boxes on given image.

draw_segmentation_masks(image, masks[, …])

Draws segmentation masks on given RGB image.

draw_keypoints(image, keypoints[, …])

Draws Keypoints on given RGB image.

flow_to_image(flow)

Converts a flow to an RGB image.

make_grid(tensor[, nrow, padding, …])

Make a grid of images.

save_image(tensor, fp[, format])

Save a given Tensor into an image file.

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