dlib-like aligning

dlib_aligning.py

import skimage
import numpy as np


LANDMARKS_SRC = np.array([[0.30392274, 0.29713595],
                          [0.68197238, 0.29713595],
                          [0.53872643, 0.57810357],
                          [0.38641786, 0.67528929],
                          [0.60154214, 0.76057500]])

def align(image, landmarks, size=(200, 200)):
    assert landmarks.shape == (5, 2)
    align_transform = skimage.transform.SimilarityTransform()
    align_transform.estimate(LANDMARKS_SRC * size, landmarks)
    image_aligned = skimage.transform.warp(image, align_transform,
        output_shape=size, preserve_range=True)
    return image_aligned