From f415df3448622f30c3d1eb680596871672b38dac Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期四, 26 十月 2017 02:51:37 +0800
Subject: [PATCH] after fg
---
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