输入关键点得到对齐人脸opencv align face

import cv2
import numpy as np
from skimage import transform as trans
# 1、输入 关键点人脸+原图 
# 2、输出 对齐后的人脸
### input img is resize square or 112 96,输入只需要关键点坐标就可以了
def process(img,landmark, align_img_size=(112,112)):

    M = None
    image_size = []
   
    src = np.array([
        [30.2946, 51.6963],
        [65.5318, 51.5014],
        [48.0252, 71.7366],
        [33.5493, 92.3655],
        [62.7299, 92.2041]], dtype=np.float32)  #this is 112 96 align img
    if align_img_size[1] == 112 and align_img_size[0] == 112:  # this is 112 112
        src[:, 0] += 8.0

    tform = trans.SimilarityTransform()
    tform.estimate(landmark, src)
    M = tform.params[0:2, :]

    # do align using landmark
    warped = cv2.warpAffine(img, M, (align_img_size[1], align_img_size[0]), borderValue=0.0)
    return warped
    
def align(landmark=None,orign_img=None):
    landmark = [62.25111822222223 ,92.23837155555555, 124.15366755555557, 96.41831111111112, 
                90.3694648888889 ,130.44167111111113 ,60.227626666666666, 153.4789884444444,
                113.25907911111112, 157.42427555555554]
    orign_img = cv2.imread("loose_crop/2.jpg")            

    landmark = np.array(landmark, dtype=np.float).reshape((5, 2))

    warped = process(orign_img,landmark) 
    bgr = warped  #[..., ::-1]

    cv2.imwrite("333.jpg", bgr)    
    
align()