From 653b3608ebe6272510b4c66f445f6f552fdc9ec9 Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期一, 11 九月 2017 05:53:10 +0800
Subject: [PATCH] Starting serious experiment without regression or iterative finetuning
---
code/test.py | 13 +++++--------
1 files changed, 5 insertions(+), 8 deletions(-)
diff --git a/code/test.py b/code/test.py
index 8e8fe50..b01d07e 100644
--- a/code/test.py
+++ b/code/test.py
@@ -27,7 +27,7 @@
default='', type=str)
parser.add_argument('--filename_list', dest='filename_list', help='Path to text file containing relative paths for every example.',
default='', type=str)
- parser.add_argument('--snapshot', dest='snapshot', help='Name of model snapshot.',
+ parser.add_argument('--snapshot', dest='snapshot', help='Path of model snapshot.',
default='', type=str)
parser.add_argument('--batch_size', dest='batch_size', help='Batch size.',
default=1, type=int)
@@ -43,7 +43,7 @@
cudnn.enabled = True
gpu = args.gpu_id
- snapshot_path = os.path.join('output/snapshots', args.snapshot + '.pkl')
+ snapshot_path = args.snapshot
# ResNet101 with 3 outputs.
# model = hopenet.Hopenet(torchvision.models.resnet.Bottleneck, [3, 4, 23, 3], 66)
@@ -58,9 +58,6 @@
model.load_state_dict(saved_state_dict)
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(),
@@ -101,9 +98,9 @@
label_roll = labels[:,2].float()
pre_yaw, pre_pitch, pre_roll, angles = model(images)
- yaw = angles[:,0].cpu().data
- pitch = angles[:,1].cpu().data
- roll = angles[:,2].cpu().data
+ yaw = angles[0][:,0].cpu().data
+ pitch = angles[0][:,1].cpu().data
+ roll = angles[0][:,2].cpu().data
# Mean absolute error
yaw_error += torch.sum(torch.abs(yaw - label_yaw) * 3)
--
Gitblit v1.8.0