如何用AI一键生成合规证件照？

import cv2
import numpy as np

# 加载预训练的模型
model = cv2.dnn.readNetFromONNX('face_alignment_model.onnx')

def generate_compliant_photo(image_path, output_path):
    # 读取图像
    image = cv2.imread(image_path)
    h, w = image.shape[:2]

    # 检测人脸
    face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
    faces = face_cascade.detectMultiScale(image, 1.1, 4)

    if len(faces) == 0:
        print("No face detected in the image.")
        return

    # 假设只有一张脸
    x, y, w, h = faces[0]
    face = image[y:y+h, x:x+w]

    # 使用模型调整人脸
    blob = cv2.dnn.blobFromImage(face, 1.0, (224, 224), (104, 117, 123), swapRB=True, crop=False)
    model.setInput(blob)
    output = model.forward()

    # 调整背景颜色为白色
    background = np.ones_like(image) * 255

    # 将调整后的脸部图像放置在白色背景上
    adjusted_face = cv2.resize(output[0], (w, h))
    background[y:y+h, x:x+w] = adjusted_face

    # 保存结果
    cv2.imwrite(output_path, background)

# 使用示例
generate_compliant_photo('input.jpg', 'output.jpg')