natanielruiz
2017-09-14 dd62d6fa4a85f18a29de009a972f5599b19ec946
code/test_AFW.py
@@ -60,7 +60,7 @@
    print 'Loading data.'
    transformations = transforms.Compose([transforms.Scale(224),
    transforms.RandomCrop(224), transforms.ToTensor()])
    transforms.CenterCrop(224), transforms.ToTensor()])
    pose_dataset = datasets.AFW(args.data_dir, args.filename_list,
                                transformations)