torchvision

tochvision是pytorch中用于图像处理的库。

torchvision由以下部分组成：

torchvision.datasets：加载数据集

torchvision.models：提供训练模型

torchvision.transforms：图形变换方法（裁剪，转向等）

    data_transform = transforms.Compose(
        [transforms.Resize(256),
         transforms.CenterCrop(224),
         transforms.ToTensor(),
         transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])])
transforms.compose将图像操作链接起来

torchvision.utils：其它方法