目录

实用程序

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

(图片、方框[、标签、...])

在给定图像上绘制边界框。

(图片, 蒙版[, ...])

在给定的 RGB 图像上绘制分割蒙版。

(图片, 关键点[, ...])

在给定的 RGB 图像上绘制关键点。

(流)

将流转换为 RGB 图像。

(tensor[, nrow, padding, ...])

制作图像网格。

(张量, fp[, 格式])

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

文档

访问 PyTorch 的全面开发人员文档

查看文档

教程

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

查看教程

资源

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

查看资源