| | |
|---|
| | | 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) |
|---|
| | |
|---|
| | | label_pitch = labels[:,1].float() |
|---|
| | | label_roll = labels[:,2].float() |
|---|
| | | |
|---|
| | | yaw, pitch, roll = model(images) |
|---|
| | | yaw, pitch, roll, angles = model(images) |
|---|
| | | |
|---|
| | | # Binned predictions |
|---|
| | | _, yaw_bpred = torch.max(yaw.data, 1) |
|---|
