From 45efc9f70f87072eb9c0b81f1771fbb99f4a1073 Mon Sep 17 00:00:00 2001
From: natanielruiz <nataniel777@hotmail.com>
Date: 星期六, 12 八月 2017 10:52:48 +0800
Subject: [PATCH] Cleanup
---
code/utils.py | 5 +++++
1 files changed, 5 insertions(+), 0 deletions(-)
diff --git a/code/utils.py b/code/utils.py
index 09a47a8..01710b2 100644
--- a/code/utils.py
+++ b/code/utils.py
@@ -7,6 +7,11 @@
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.
--
Gitblit v1.8.0