| | |
|---|
| | | 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] |
|---|
| | |
|---|
| | | 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 |
|---|
| | |
|---|
| | | cv2.line(img, (int(x3), int(y3)), (int(x3+x2-face_x),int(y3+y2-face_y)),(0,255,0),2) |
|---|
| | | |
|---|
| | | return img |
|---|
| | | |
|---|
| | | def draw_axis(img, yaw, pitch, roll, tdx=None, tdy=None, size = 100): |
|---|
| | | |
|---|
| | | pitch = pitch * np.pi / 180 |
|---|
| | | yaw = -(yaw * np.pi / 180) |
|---|
| | | roll = roll * np.pi / 180 |
|---|
| | | |
|---|
| | | if tdx != None and tdy != None: |
|---|
| | | tdx = tdx |
|---|
| | | tdy = tdy |
|---|
| | | else: |
|---|
| | | height, width = img.shape[:2] |
|---|
| | | tdx = width / 2 |
|---|
| | | tdy = height / 2 |
|---|
| | | |
|---|
| | | # X-Axis pointing to right. drawn in red |
|---|
| | | x1 = size * (cos(yaw) * cos(roll)) + tdx |
|---|
| | | y1 = size * (cos(pitch) * sin(roll) + cos(roll) * sin(pitch) * sin(yaw)) + tdy |
|---|
| | | |
|---|
| | | # Y-Axis | drawn in green |
|---|
| | | # v |
|---|
| | | x2 = size * (-cos(yaw) * sin(roll)) + tdx |
|---|
| | | y2 = size * (cos(pitch) * cos(roll) - sin(pitch) * sin(yaw) * sin(roll)) + tdy |
|---|
| | | |
|---|
| | | # Z-Axis (out of the screen) drawn in blue |
|---|
| | | x3 = size * (sin(yaw)) + tdx |
|---|
| | | y3 = size * (-cos(yaw) * sin(pitch)) + tdy |
|---|
| | | |
|---|
| | | cv2.line(img, (int(tdx), int(tdy)), (int(x1),int(y1)),(0, 255, 255),3) |
|---|
| | | cv2.line(img, (int(tdx), int(tdy)), (int(x2),int(y2)),(0,0,255),3) |
|---|
| | | cv2.line(img, (int(tdx), int(tdy)), (int(x3),int(y3)),(0,255,0),3) |
|---|
| | | |
|---|
| | | return img |
|---|
