Pytorch - Augmentation

[junxnone]

Reference

• Pytorch 中的数据增强方式最全解释

Brief

from torchvision import transforms

lambd = lambda x: TF.rotate(x, 100)
transform = transforms.Compose([
    transforms.ToTensor(),
    transforms.Normalize(mean, std),
    transforms.Resize(size),
    transforms.CenterCrop(size),
    transforms.Pad(padding, padding_mode="symmetric"),
    transforms.Lambda(lambd),
    transforms.ToPILImage()
])

tran_img = transform(img)

[junxnone]

311