Pytorch学习之旅(6)——transforms类

一、transforms()类

torchvision.transforms()：用于常见的一些图形变换.

链接：transforms()类的常用方法

二、使用

1、运用transforms.Compose()函数，将多个transforms的子类函数拼接成list()：

# 串联图片变换的多个操作
valid_transform = transforms.Compose([
    transforms.Resize((32, 32)),
    transforms.ToTensor(),
    transforms.Normalize(norm_mean, norm_std)
])	# 返回list()列表，元素为多个图形变换函数

2、在实例化dataset子类，将返回的列表设置为初始值：

# 构建Dataset
train_data = RMBDataset(data_dir=train_dir, transform=valid_transform ) # RMBDataset为Dataset的子类.