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