keshik6

# 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)
                                          ]),