import cv2
import dlib
# μΌκ΅΄ κ²μΆκΈ°μ λλλ§ν¬ κ²μΆκΈ° μμ± --- β
detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor('./shape_predictor_68_face_landmarks.dat')
img = cv2.imread("../img/man_face.jpg")
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# μΌκ΅΄ μμ κ²μΆ --- β‘
faces = detector(gray)
for rect in faces:
# μΌκ΅΄ μμμ μ’νλ‘ λ³ν ν μ¬κ°ν νμ --- β’
x,y = rect.left(), rect.top()
w,h = rect.right()-x, rect.bottom()-y
cv2.rectangle(img, (x, y), (x + w, y + h), (0, 255, 0), 1)
# μΌκ΅΄ λλλ§ν¬ κ²μΆ --- β£
shape = predictor(gray, rect)
for i in range(68):
# λΆμλ³ μ’ν μΆμΆ λ° νμ --- β€
part = shape.part(i)
cv2.circle(img, (part.x, part.y), 2, (0, 0, 255), -1)
cv2.putText(img, str(i), (part.x, part.y), cv2.FONT_HERSHEY_PLAIN, \
0.5,(255,255,255), 1, cv2.LINE_AA)
cv2.imshow("face landmark", img)
cv2.waitKey(0)