add convert_98_to_68_landmarks

Author: xinntao

facexlib/alignment/convert_98_to_68_landmarks.py

@@ -0,0 +1,82 @@
+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