Openface（二）：实现人脸对齐

最新推荐文章于 2025-11-06 01:19:44 发布

原创

最新推荐文章于 2025-11-06 01:19:44 发布 · 2.7k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#openface

本文介绍了如何利用开源库Openface进行人脸对齐，重点在于将原始代码简化并利用fire模块处理参数，同时提到了dlib库在人脸对齐过程中的作用。通过对LFW数据集的应用，这些对齐后的图片可以作为深度学习模型的训练数据。

最近学习了开源代码Openface，实现了LFW数据集的人脸对齐。这些人脸对齐的图片是后续深度学习神经网络的训练数据集。
代码除了引用常见的python包，还引用了两个本地包：一个是dataset，用以生成图像generator对象和辅助输出；另一个是dlib_align，是上一篇文章中的使用dlib库对齐人脸的包。

本代码将原代码中的parse换为了更为简洁的fire，使用Config类管理对齐函数所需的输入。在实际调用时，只需要在terminal中输入

python align.py alignMain --size=100

即可更新参数。

代码及注释如下：

# align script
# 使用dlib对齐

import cv2
import numpy as np
import os
import sys
import random
import shutil
import fire

# 提取本文件所在目录
fileDir = os.path.dirname(os.path.realpath(__file__))
# 在flidDir同级目录中寻找models文件夹，找到dlib参数的路径
modelDir = os.path.join(fileDir, '..', 'models') 
dlibModelDir = os.path.join(modelDir, 'dlib') #这是存放dilb对齐模板标记点的地方
myfaceDir = os.path.join(modelDir, 'myface')