torchvison图像处理详解

性感官方，在线文档

transforms.Compose

串联各种图像处理方法，依次进行各个图像处理方法。

transforms.Compose(
            [
                transforms.Resize(size = (224,224)),#尺寸规范
                transforms.ToTensor(),   #转化为tensor
                transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
            ])

transforms.CenterCrop

图片裁剪，以图片中心为中心。

torchvision.transforms.CenterCrop(size = size)

这里的size可以是整数，裁剪后的图片是正方形，也可以是[h,w]，表示长和宽。

transforms.ColorJitter

将图片随机改变颜色。暂时我还用不到，先不写了。

torchvision.transforms.ColorJitter(brightness=0, contrast=0, saturation=0, hue=0)

transforms.FiveCrop

输出图片的crop结果，五张图片，分别是中心裁剪一张，和四角裁剪四张。

torchvision.transforms.FiveCrop(size)

size和centercrop中的size一样。

transforms.Grayscale

图片灰度化。

torchvision.transforms.Grayscale(num_output_channels=1)

num_output_channels可选1和3,1表示灰度化，3表示输出r=g=b的图片。

transforms.Resize

resize

torchvision.transforms.Resize(size, interpolation=2)

size和之前的size一样，interpolation表示插值方法，暂时我还没有用到，这块先不写。

transforms.Normalize

数据标准化

torchvision.transforms.Normalize(mean, std, inplace=False)

标准化公式为(value - mean)/std，inplace默认为false。注意mean和std的格式为(mean[1],…,mean[n])，，并且使用该方法之前要totenser。

transforms.ToTensor

将图片转化为tensor

transforms.ToTensor()