实用工具

torchvision.utils 模块包含各种实用程序，主要用于可视化。

draw_bounding_boxes(image, boxes[, labels, ...])	在给定的 RGB 图像上绘制边界框。
draw_segmentation_masks(image, masks[, ...])	在给定的 RGB 图像上绘制分割掩码。
draw_keypoints(image, keypoints[, ...])	在给定的 RGB 图像上绘制关键点。
flow_to_image(flow)	将流转换为 RGB 图像。
make_grid(tensor[, nrow, padding, ...])	制作图像网格。
save_image(tensor, fp[, format])	将给定的张量保存到图像文件中。