人脸匹配对齐算法pytorch_从零开始搭建人脸识别系统（二）：人脸对齐

从零开始搭建人脸识别系统(一)MTCNN 中我们讲了如何在一张图片中定位人脸框的位置。设想这样一种情况，图片中的脸相对于图片是斜的:(下面的图由于人脸关键点比较小可能看不清楚，可以打开原图可以看到标识的关键点)。

import cv2

import matplotlib.pyplot as plt

import numpy as np

img_file = '../tests/asset/images/roate.jpg'

img = cv2.imread(img_file)

img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)

plt.figure(figsize=(10, 10))

plt.imshow(img)

plt.show()

这时我们需要把图片中的脸扶正，再送入人脸识别模块。那我们要怎么做呢，我们结合代码一步一步地来讲。本文的代码同样是基于faciallab/FaceDetector​github.com

如果本文对你有帮助可以帮我点个星星哦。

首先我们加载mtcnn模型

import mtcnn

from mtcnn.utils import draw

# First we create pnet, rnet, onet, and load weights from caffe model.

pnet, rnet, onet = mtcnn.get_net_caffe('../output/converted')

# Then we create a detector

<