Shortcuts

工具

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])

将给定的Tensor保存为图像文件。