Python 进行人脸校正

2019-03-28 12:05:36 浏览数 (1)

已同步:博客 | 公众号 | 语雀

问题描述

在做人脸识别的时候,前期的数据处理过程中可能会遇到一个问题,即将人脸从不同尺寸的图像中截取出来,再进行“对齐”操作。这样可以使每一个截取的人脸中的眼睛等位置处于同一位置,会对后面的识别算法起到一定的优化作用。

比如,下面 3 张图片所示,人脸的位置,图像的大小各不一样。我们所需要做的就是,将人脸从各个图片中截取出来,再旋转进行校正,使得眼睛在同一位置上,最后将图片的大小也统一调整为 224 x 224。

校正的效果如下图所示:

实现步骤

前提:已经提取出每张图片里眼睛的坐标,只需要读取数据就行

整个过程遵循以下几个步骤:

  • 找两眼间的直线距离并计算该直线与水平线之间的夹角,即倾斜角度
  • 根据找到的倾斜角度旋转图片
  • 在旋转后的图片中找到眼睛的位置
  • 根据眼睛坐标找到包含面部的框的宽度和高度
  • 调整图片的尺寸
1. 计算直线距离及倾斜角度

计算两眼之间的距离很简单,只要找出两个向量的坐标就可以计算出来。计算两眼之间的距离直线的倾斜角度,就是将两个向量相减,得到一组横纵坐标,然后利用 arctan 的公式求解角度,公式如下所示:

大多数 sin / cos 等函数要求角度为弧度,可以使用这个转换公式:angle = angle * 180 / π。

代码语言:javascript复制
"""
1.找到眼睛倾斜的角度和两眼距离
"""
p1 = np.array(eye[0])[::-1] # 左眼坐标
p2 = np.array(eye[1])[::-1] # 右眼坐标
dist = np.sqrt(np.sum(p1-p2)**2) # 两只眼睛之间的距离

dp = p1 - p2
angle = np.arctan(dp[0] / dp[1])
2. 根据找到的角度旋转图片

上一步骤中,求出了眼睛的倾斜角度,这一步就可以使用该角度来旋转图片了。这里方便地调用 scipy.ndimage.rotate 来旋转图片。之后再计算出旋转后图片的中点。

代码语言:javascript复制
"""
2. 旋转图片
"""
from scipy import ndimage

# 旋转图片
rot_img = ndimage.rotate(face, angle= angle*180/np.pi)
# 旋转后图像的中点
rot_image_center = np.array((np.array(rot_img.shape[:2]) - 1) / 2, 
dtype=np.int)
3. 寻找旋转后眼睛的位置

寻找旋转后眼睛的位置,就是寻找旋转后两个点的坐标。旋转的时候,我们是以图片中点为原点来旋转的,因而需要先求解出原图片的原点,也就是图片的中心点。下面是以原点 (x0, y0) 为中心旋转后,求坐标的公式:

代码语言:javascript复制
"""
3. 在旋转后的图片中找到眼睛的坐标
"""

# 原两眼距离的中点
org_eye_center = np.array((p1   p2) / 2, dtype=np.int)
# 原图像的中点
org_image_center = np.array((np.array(face.shape[:2]) - 1) / 2, dtype=np.int)
# 以图片中心进行旋转,在旋转后的图片中找到眼睛的中点
R = np.array([[np.cos(angle), np.sin(angle)], [-np.sin(angle), np.cos(angle)]])
rot_eye_center = np.dot(R, org_eye_center[::-1]
                        -org_image_center[::-1])[::-1]   rot_image_center
rot_eye_center = np.array(rot_eye_center, dtype=int)
4. 根据眼睛坐标找到包含面部的框的宽度和高度

找到旋转后眼睛的坐标,可以利用眼睛的坐标来框出人脸的大致范围。这里就直接看代码,从代码中比较好理解。

代码语言:javascript复制
"""
4. 根据眼睛坐标找到包含面部的框的宽度和高度
"""
mid_y, mid_x = rot_eye_center
    
MUL = 2
y_top = int(max(mid_y - MUL * dist, 0))
y_bot = int(min(mid_y   MUL * dist, rot_img.shape[0]))
x_left = int(max(mid_x - MUL * dist, 0))
x_right = int(min(mid_x   MUL * dist, rot_img.shape[1]))
    
cropped_img = rot_img[y_top:y_bot 1, x_left:x_right 1, :]
5. 裁剪图片

将旋转并框出的人脸进行裁剪,将其设定为统一的尺寸。

代码语言:javascript复制
"""
5. 裁剪图像的尺寸为 224x224
"""
import skimage.transform as transform

scaled_img = transform.resize(cropped_img, [224, 224], mode='constant')
plt.imshow(scaled_img)

完整的代码链接:face-alignment

参考文献

[1]. Coursera - Deep Learning in Computer Vision Week 1 assigment [2]. 图像旋转的原理,实现与优化 [3]. Rotated image coordinates after scipy.ndimage.interpolation.rotate?

0 人点赞