Omg

natanielruiz

2017-09-07 6dd2ff502947ec809d420e2baefa023d821a8bb1

 code/test.py

@@ -59,8 +59,12 @@

    print 'Loading data.'

    # transformations = transforms.Compose([transforms.Scale(224),
    # transforms.RandomCrop(224), transforms.ToTensor()])

    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.AFLW2000(args.data_dir, args.filename_list,
                                transformations)