You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
2.9 KiB
Python
83 lines
2.9 KiB
Python
import numpy as np
|
|
|
|
|
|
def load_txt_file(file_path):
|
|
"""Load data or string from txt file."""
|
|
|
|
with open(file_path, 'r') as cfile:
|
|
content = cfile.readlines()
|
|
cfile.close()
|
|
content = [x.strip() for x in content]
|
|
num_lines = len(content)
|
|
return content, num_lines
|
|
|
|
|
|
def anno_parser(anno_path, num_pts, line_offset=0):
|
|
"""Parse the annotation.
|
|
Args:
|
|
anno_path: path of anno file (suffix .txt)
|
|
num_pts: number of landmarks.
|
|
line_offset: first point starts, default: 0.
|
|
|
|
Returns:
|
|
pts: num_pts x 2 (x, y)
|
|
"""
|
|
|
|
data, _ = load_txt_file(anno_path)
|
|
n_points = num_pts
|
|
# read points coordinate.
|
|
pts = np.zeros((n_points, 2), dtype='float32')
|
|
for point_index in range(n_points):
|
|
try:
|
|
pts_list = data[point_index + line_offset].split(',')
|
|
pts[point_index, 0] = float(pts_list[0])
|
|
pts[point_index, 1] = float(pts_list[1])
|
|
except ValueError:
|
|
print(f'Error in loading points in {anno_path}')
|
|
return pts
|
|
|
|
|
|
def landmark_98_to_68(landmark_98):
|
|
"""Transfer 98 landmark positions to 68 landmark positions.
|
|
Args:
|
|
landmark_98(numpy array): Polar coordinates of 98 landmarks, (98, 2)
|
|
Returns:
|
|
landmark_68(numpy array): Polar coordinates of 98 landmarks, (68, 2)
|
|
"""
|
|
|
|
landmark_68 = np.zeros((68, 2), dtype='float32')
|
|
# cheek
|
|
for i in range(0, 33):
|
|
if i % 2 == 0:
|
|
landmark_68[int(i / 2), :] = landmark_98[i, :]
|
|
# nose
|
|
for i in range(51, 60):
|
|
landmark_68[i - 24, :] = landmark_98[i, :]
|
|
# mouth
|
|
for i in range(76, 96):
|
|
landmark_68[i - 28, :] = landmark_98[i, :]
|
|
# left eyebrow
|
|
landmark_68[17, :] = landmark_98[33, :]
|
|
landmark_68[18, :] = (landmark_98[34, :] + landmark_98[41, :]) / 2
|
|
landmark_68[19, :] = (landmark_98[35, :] + landmark_98[40, :]) / 2
|
|
landmark_68[20, :] = (landmark_98[36, :] + landmark_98[39, :]) / 2
|
|
landmark_68[21, :] = (landmark_98[37, :] + landmark_98[38, :]) / 2
|
|
# right eyebrow
|
|
landmark_68[22, :] = (landmark_98[42, :] + landmark_98[50, :]) / 2
|
|
landmark_68[23, :] = (landmark_98[43, :] + landmark_98[49, :]) / 2
|
|
landmark_68[24, :] = (landmark_98[44, :] + landmark_98[48, :]) / 2
|
|
landmark_68[25, :] = (landmark_98[45, :] + landmark_98[47, :]) / 2
|
|
landmark_68[26, :] = landmark_98[46, :]
|
|
# left eye
|
|
LUT_landmark_68_left_eye = [36, 37, 38, 39, 40, 41]
|
|
LUT_landmark_98_left_eye = [60, 61, 63, 64, 65, 67]
|
|
for idx, landmark_98_index in enumerate(LUT_landmark_98_left_eye):
|
|
landmark_68[LUT_landmark_68_left_eye[idx], :] = landmark_98[landmark_98_index, :]
|
|
# right eye
|
|
LUT_landmark_68_right_eye = [42, 43, 44, 45, 46, 47]
|
|
LUT_landmark_98_right_eye = [68, 69, 71, 72, 73, 75]
|
|
for idx, landmark_98_index in enumerate(LUT_landmark_98_right_eye):
|
|
landmark_68[LUT_landmark_68_right_eye[idx], :] = landmark_98[landmark_98_index, :]
|
|
|
|
return landmark_68
|