Pytorch torchvision包介绍

torchvision 是PyTorch中专门用来处理图像的库。这个包中有四个大类。

torchvision.datasets 用来进行数据加载的，PyTorch团队在这个包中帮我们提前处理好了很多很多图片数据集。

torchvision.models 提供已经训练好的模型，让我们可以加载之后，直接使用。

我们可以直接使用如下代码来快速创建一个权重随机初始化的模型：

import torchvision.models as models
resnet18 = models.resnet18()
alexnet = models.alexnet()
squeezenet = models.squeezenet1_0()

也可以通过使用 pretrained=True 来加载一个别人预训练好的模型

import torchvision.models as models
resnet18 = models.resnet18(pretrained=True)
alexnet = models.alexnet(pretrained=True)

torchvision.transforms 常用的图片变换，例如裁剪、旋转等

torchvision.utils 其他的一些有用的方法