New experiments with hourglass

natanielruiz

2017-08-15 92ed4cb2ea68be44b1ff153e00410c2082ee62df

 code/train.py

@@ -105,7 +105,7 @@
    transformations = transforms.Compose([transforms.Scale(224),transforms.RandomCrop(224),
                                          transforms.ToTensor()])

    pose_dataset = datasets.Pose_300W_LP_binned(args.data_dir, args.filename_list,
    pose_dataset = datasets.Pose_300W_LP(args.data_dir, args.filename_list,
                                transformations)
    train_loader = torch.utils.data.DataLoader(dataset=pose_dataset,
                                               batch_size=batch_size,