作用:transforms.Compose()类用来组合多个torchvision.transforms操作。
参数:一个list数组,数组里是多个'Transform'对象,即[transforms, transforms...]。
操作:如下所示,遍历list数组,对img依次执行每个transforms操作,并返回transforms后的img。
def __call__(self, img):
for t in self.transforms:
img = t(img)
return img
示例:
transforms.Compose([
transforms.CenterCrop(10),
transforms.ToTensor(),
])