针对FFHQFaceAlignment只能裁剪对齐单个人脸的改进，使用dlib库

概述

https://blog.csdn.net/yue_la/article/details/140404258在这篇文章中，我写了FFHQFaceAlignment的代码解读，但是它只能裁剪对齐单个人脸，针对多人脸图像，只能检测一个人脸。我写的agent想实现一个功能：可以针对多人图像中的单个人脸进行裁剪对齐，添加一个参数face_index来进行人脸选择。所以编写了这个代码，原代码中的模型并不支持多人脸检测，改用dlib库和相关模型。

dlib库安装

linux系统可以使用下面的命令安装dlib库

pip install cmake
pip install boost
pip install dlib

如果安装失败，可以尝试使用conda安装

conda install -c conda-forge dlib

这行命令需要等待比较长时间，接近十分钟，按照提示选择y + 回车

代码

导库

import os
import os.path as osp
from tqdm import tqdm
import torch
import numpy as np
import cv2
import PIL.Image
import PIL.ImageFile
from PIL import Image
import scipy.ndimage
import dlib

align_crop_image函数

def align_crop_image(self, image, landmarks, transform_size=256):
        lm = landmarks
        lm_chin = lm[0: 17]
        lm_eyebrow_left = lm[17: 22]
        lm_eyebrow_right = lm[22: 27]
        lm_nose = lm[27: 31]
        lm_nostrils = lm[31: 36]
        lm_eye_left = lm[36: 42]
        lm_eye_right = lm[42: 48]
        lm_mouth_outer = lm[48: 60]
        lm_mouth_inner = lm[60: 68]

        eye_left = np.mean(lm_eye_left, axis=0)
        eye_right = np.mean(l