Center crop instead of random crop for testing.

natanielruiz

2017-09-13 dfa3664a0f56445b023020a0ddb5eedc2780169a

Center crop instead of random crop for testing.

 code/batch_testing.py

@@ -62,7 +62,7 @@
    print 'Loading data.'

    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])])

    if args.dataset == 'AFLW2000':

 code/batch_testing_preangles.py

@@ -61,7 +61,7 @@
    print 'Loading data.'

    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])])

    if args.dataset == 'AFLW2000':

 code/hopenet.py

@@ -120,7 +120,7 @@
        angles.append(torch.cat([yaw, pitch, roll], 1))

        for idx in xrange(self.iter_ref):
            angles.append(self.fc_finetune(torch.cat((angles[-1], x), 1)))
            angles.append(self.fc_finetune(torch.cat((angles[idx], x), 1)))

        return pre_yaw, pre_pitch, pre_roll, angles


 code/test.py

@@ -62,7 +62,7 @@
    print 'Loading data.'

    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])])

    if args.dataset == 'AFLW2000':

 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)

 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,

 code/test_on_video.py

@@ -60,7 +60,7 @@
    print 'Loading data.'

    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])])

    model.cuda(gpu)

 code/test_preangles.py

@@ -64,7 +64,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])])

    if args.dataset == 'AFLW2000':

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