目录

实用工具

The torchvision.utils 模块包含各种实用工具,主要用于可视化

draw_bounding_boxes(image, boxes[, labels, ...])

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

draw_segmentation_masks(image, masks[, ...])

在给定的RGB图像上绘制分割掩码。

draw_keypoints(image, keypoints[, ...])

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

flow_to_image(流程)

将流转换为RGB图像。

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

制作图像网格。

save_image(tensor, fp[, format])

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

文档

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

查看文档

教程

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

查看教程

资源

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

查看资源