transforms.Compose()类的使用说明

作用: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(),
])

 

你可能感兴趣的:(pytorch)