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