face_landmarks.py

# -*- coding: utf-8 -*-
"""
Created on Thu Jun  8 14:35:41 2023

@author: slbouknight
"""

import face_recognition
from PIL import Image, ImageDraw

# Load image into numpy array
face_image = face_recognition.load_image_file('images/samples/rihanna.jpg')

# Obtain list of face landmarks
face_landmarks_list = face_recognition.face_landmarks(face_image)

# Display landmarks
print(face_landmarks_list)

# Iterate through all face landmarks
for face_landmarks in face_landmarks_list:
    # Convert np array to pil image and make a DrawObject
    pil_image = Image.fromarray(face_image)
    d = ImageDraw.Draw(pil_image)
    
    # Connect each face landmark point using a white line
    d.line(face_landmarks['chin'], fill=(0, 255, 0), width=2)
    d.line(face_landmarks['left_eyebrow'], fill=(0, 255, 0), width=2)
    d.line(face_landmarks['right_eyebrow'], fill=(0, 255, 0), width=2)
    d.line(face_landmarks['nose_bridge'], fill=(0, 255, 0), width=2)
    d.line(face_landmarks['nose_tip'], fill=(0, 255, 0), width=2)
    d.line(face_landmarks['left_eye'], fill=(0, 255, 0), width=2)
    d.line(face_landmarks['right_eye'], fill=(0, 255, 0), width=2)
    d.line(face_landmarks['top_lip'], fill=(0, 255, 0), width=2)
    d.line(face_landmarks['bottom_lip'], fill=(0, 255, 0), width=2)

# Show final image
pil_image.show()

# Save image
pil_image.save('images/samples/abhi_landmarks.jpg')