Utils — Torchvision 0.22 documentation (original) (raw)
The torchvision.utils module contains various utilities, mostly for visualization.
| draw_bounding_boxes(image, boxes[, labels, ...]) | Draws bounding boxes on given RGB 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. |