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 其他的一些有用的方法

标签: torchvision包, pytorch

添加新评论