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.