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.