From 1ddaaf685a3c5febca32305fc588f982b5e4cdaa Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期四, 28 九月 2017 06:21:15 +0800
Subject: [PATCH] Final code before submission
---
code/hopenet.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/code/hopenet.py b/code/hopenet.py
index de2f4ec..80160d9 100644
--- a/code/hopenet.py
+++ b/code/hopenet.py
@@ -339,7 +339,7 @@
preangles = torch.cat([yaw, pitch, roll], 1)
angles.append(preangles)
- return pre_yaw, pre_pitch, pre_roll, angles, sr_output
+ return pre_yaw, pre_pitch, pre_roll, angles, sr_y
class Hopenet_new(nn.Module):
# This is just Hopenet with 3 output layers for yaw, pitch and roll.
--
Gitblit v1.8.0