alignment/example.py at master · lebedevdes/alignment · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
import face_alignment
from skimage import io
import cv2
import time
import pickle
import face_preprocess
import numpy as np

def get_center_line_point(p1, p2):
    x = p1[0] + (p2[0] - p1[0]) / 2
    y = p1[1] + (p2[1] - p1[1]) / 2
    return [x, y]


fa = face_alignment.FaceAlignment(face_alignment.LandmarksType._2D, flip_input=False, device='cpu')

m = fa.face_alignment_net.eval()
print(m)


input_image = cv2.imread('images/Tom_Hanks_54745.png')

# start = time.time()
# preds = fa.get_landmarks(input_image)
# end = time.time()

# with open('landmarks.pickle', 'wb') as f:
#     pickle.dump(preds, f)

with open('landmarks.pickle', 'rb') as f:
    preds = pickle.load(f)

# for i in preds[0]:
#     cv2.circle(input_image, (int(i[0]), int(i[1])), 3, (0, 0, 0), -1)

nose = preds[0][30]
r_right_eye = preds[0][36]
l_right_eye = preds[0][39]
c_right_eye = get_center_line_point(r_right_eye, l_right_eye)
r_left_eye = preds[0][42]
l_left_eye = preds[0][45]
c_left_eye = get_center_line_point(r_left_eye, l_left_eye)
right_month = preds[0][48]
left_month = preds[0][54]
# chin = preds[0][8]

# p = preds[0][8]

# landmarks = [right_eye, left_eye, nose, right_month, left_month, chin]
landmarks = np.array([c_right_eye, c_left_eye, nose, right_month, left_month])
for i in landmarks:
    print(i)
    cv2.circle(input_image, (int(i[0]), int(i[1])), 3, (0, 0, 255), -1)

# print(end - start)


# cv2.imshow("test", input_image)
# cv2.waitKey()
# cv2.destroyAllWindows()

# print(preds)

# alignmented = face_preprocess.preprocess(input_image, landmark=landmarks, image_size='112,112')

# print(alignmented.shape)

cv2.imshow("test2", input_image)
cv2.waitKey()
cv2.destroyAllWindows()