包含流行的数据集,模型架构以及一些用于计算机视觉的常见的图像转换函数。
torchvision.datasets
torchvision.models
torchvision.transforms
注意,传入的图片需为 uint8 类型。
uint8
torchvision.utils
← PyTorch Tensor PyTorch 使用技巧 →