From 31fc66b795c0a57b8009d7b03f49f6cd099ceb29 Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期六, 23 九月 2017 12:07:48 +0800
Subject: [PATCH] Trying superres

---
 code/utils.py |    8 ++++++--
 1 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/code/utils.py b/code/utils.py
index 09a47a8..96c447b 100644
--- a/code/utils.py
+++ b/code/utils.py
@@ -7,9 +7,14 @@
 import math
 from math import cos, sin
 
+def softmax_temperature(tensor, temperature):
+    result = torch.exp(tensor / temperature)
+    result = torch.div(result, torch.sum(result, 1).unsqueeze(1).expand_as(result))
+    return result
+
 def get_pose_params_from_mat(mat_path):
     # This functions gets the pose parameters from the .mat
-    # Annotations that come with the 300W_LP dataset.
+    # Annotations that come with the Pose_300W_LP dataset.
     mat = sio.loadmat(mat_path)
     # [pitch yaw roll tdx tdy tdz scale_factor]
     pre_pose_params = mat['Pose_Para'][0]
@@ -45,7 +50,6 @@
     p = pitch * np.pi / 180
     y = -(yaw * np.pi / 180)
     r = roll * np.pi / 180
-
     if tdx != None and tdy != None:
         face_x = tdx - 0.50 * size
         face_y = tdy - 0.50 * size

--
Gitblit v1.8.0