code/test_old.py
@@ -63,7 +63,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])]) pose_dataset = datasets.AFLW2000(args.data_dir, args.filename_list,