natanielruiz
2017-09-11 653b3608ebe6272510b4c66f445f6f552fdc9ec9
code/test_AFLW.py
@@ -60,7 +60,8 @@
    print 'Loading data.'
    transformations = transforms.Compose([transforms.Scale(224),
    transforms.RandomCrop(224), transforms.ToTensor()])
    transforms.RandomCrop(224), transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])])
    pose_dataset = datasets.AFLW(args.data_dir, args.filename_list,
                                transformations)