Add geometric transformation data to my training Set

1

Hello,

I have a dataset of 100 labeled data from the CIFAR dataset.
I would like to do some geometric transformation to this data, and add it to the training set . What is the easiest way to do it??

Thanks !!

2

Hello,

You could add transformations by using torchvision.transforms here.

import torchvision.transforms as transforms
transforms_ = transforms.Compose(
    transforms.ToTensor(),
    ...
)

And then create a DataLoader pass the dataset and transforms_ to the dataloader.