From 8736b9753604a2e88843ba87c7e0e688dce072e6 Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期五, 07 七月 2017 15:12:22 +0800
Subject: [PATCH] Next
---
code/test_resnet_bins.py | 3 +++
1 files changed, 3 insertions(+), 0 deletions(-)
diff --git a/code/test_resnet_bins.py b/code/test_resnet_bins.py
index f5be4f8..30aa158 100644
--- a/code/test_resnet_bins.py
+++ b/code/test_resnet_bins.py
@@ -46,6 +46,7 @@
# ResNet50 with 3 outputs.
model = hopenet.Hopenet(torchvision.models.resnet.Bottleneck, [3, 4, 6, 3], 66)
+ # model = hopenet.Hopenet(torchvision.models.resnet.BasicBlock, [2, 2, 2, 2], 66)
print 'Loading snapshot.'
# Load snapshot
@@ -107,6 +108,8 @@
pitch_error += abs(pitch_predicted - label_pitch[0]) * 3
roll_error += abs(roll_predicted - label_roll[0]) * 3
+ # print yaw_predicted * 3, label_yaw[0] * 3, abs(yaw_predicted - label_yaw[0]) * 3
+
# for er in xrange(0,n_margins):
# yaw_correct[er] += (label_yaw[0] in range(yaw_predicted[0,0] - er, yaw_predicted[0,0] + er + 1))
# pitch_correct[er] += (label_pitch[0] in range(pitch_predicted[0,0] - er, pitch_predicted[0,0] + er + 1))
--
Gitblit v1.8.0