Train time Transformations for pascal VOC

transformations.py

# This is for model 1
transformations = transforms.Compose([transforms.Resize(330),
                                     transforms.RandomChoice([
                                             transforms.CenterCrop(300),
                                             transforms.RandomResizedCrop(300, scale=(0.80, 1.0)),
                                             ]),                                      
                                      transforms.RandomChoice([
                                          transforms.ColorJitter(brightness=(0.80, 1.20)),
                                          transforms.RandomGrayscale(p = 0.25)
                                          ]),
                                      transforms.RandomHorizontalFlip(p = 0.25),
                                      transforms.RandomRotation(25),
                                      transforms.ToTensor(), 
                                      transforms.Normalize(mean = mean, std = std),
                                      ])


# This is for model 2
transformations = transforms.Compose([transforms.Resize((300, 300)),                                     
                                      transforms.RandomChoice([
                                          transforms.ColorJitter(brightness=(0.80, 1.20)),
                                          transforms.RandomGrayscale(p = 0.25)
                                          ]),
                                      transforms.RandomHorizontalFlip(p = 0.25),
                                      transforms.RandomRotation(25),
                                      transforms.ToTensor(), 
                                      transforms.Normalize(mean = mean, std = std),
                                      ])

# The purpose of model 1 was to add noise labels during training to see whether the model becomes robust.
# The rest of the comman augmentations are for robustness as well.