快捷方式

实用程序

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

将给定的张量保存到图像文件中。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源