pytorch中的transform函数

torchvision.transforms是pytorch中的图像预处理包.
一般用Compose把多个步骤整合到一起:

transforms.Compose([
transforms.CenterCrop(10),
transforms.ToTensor(),
])

这样就把两个步骤整合到一起,接下来是transforms中的函数:
Resize把给定的图片resize到given size
Normalize就是将一个tensor image根据其均值和方差进行归一化
ToTensor就是将一个PIL image转换成一个Tensor,PIL是(HWC)的形式,范围是[0,255],Tensor是(CHW)的形式,范围是[0,1]。
ToPILImage是将一个tensor转换成PIL image
CenterCrop在图片的中间区域进行裁剪
RandomCrop在一个随机的位置进行裁剪
RandomHorizontalFlip是以0.5的概率水平翻转给定的PIL图像
RandomVerticalFlip是以0.5的概率竖直翻转给定的PIL图像
RandomResizedCrop将PIL图像裁剪成任意大小和纵横比
Grayscale将图像转换为灰度图像
RandomGrayscale将图像以一定的概率转换为灰度图像
FiceCrop把图像裁剪为四个角和一个中心
pad填充
ColorJitter随机改变图像的亮度对比度和饱和度

你可能感兴趣的:(机器学习)