From 8f2b081586161a55388934456a9b0193d02dd267 Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期三, 27 九月 2017 10:15:27 +0800
Subject: [PATCH] Random downsample experiment
---
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