Example of using AWS Rekognition to identify and draw bounding boxes for faces

example-img-bbox.jpg

example-img.jpg

main.py

"""Use AWS Rekognition face detection to produce bounding boxes"""

import boto3
from PIL import Image, ImageDraw


IMG = 'example-img.jpg'
REGION = 'us-east-1'


def get_face_details(img_path):

    # https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/rekognition.html#Rekognition.Client.detect_faces

    rekognition = boto3.client("rekognition", REGION)

    with open(img_path, "rb") as image_file:
        encoded_img = image_file.read()
        response = rekognition.detect_faces(
            Image={
                'Bytes': encoded_img
            }
        )
    return response['FaceDetails']


def main():
    face_details = get_face_details(IMG)

    im = Image.open(IMG)
    (width, height) = im.size
    draw = ImageDraw.Draw(im)

    for idx, face in enumerate(face_details):
        print('Face Detected {} with confidence {:.4f}'.format(idx, face['Confidence']))

        bbox = face['BoundingBox']
        x1, y1 = (bbox['Left'] * width, bbox['Top'] * height)
        x2, y2 = (x1 + bbox['Width'] * width, y1 + bbox['Height'] * height)
        draw.rectangle(xy=[x1, y1, x2, y2], outline='red', width=3)

    im.save('example-img-bbox.jpg', 'JPEG')


if __name__ == '__main__':
    main()

requirements.txt

Pillow
boto3