code/test_preangles.py
@@ -64,7 +64,7 @@ # transforms.RandomCrop(224), transforms.ToTensor()]) transformations = transforms.Compose([transforms.Scale(224), transforms.RandomCrop(224), transforms.ToTensor(), transforms.CenterCrop(224), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])]) if args.dataset == 'AFLW2000':