摘要
本发明涉及一种基于小波扩散模型的人脸数据合成方法,属于数字图像处理领域。该方法包括:S1:数据准备阶段;S2:在每一轮训练前需采样身份图像并提取图像的身份特征,使用人脸识别模型提取人脸图像的身份特征,然后将该特征进行归一化和线性处理;S3:从风格库中随机采样若干张风格图像,然后使用风格编码块对其进行编码;S4:每轮训练中,先从身份数据中选择一张图像作为模型输入和身份图像,然后对该图像逐步加噪,加噪完成后模型开始逐步去噪以还原图像;S5:模型训练开始前随机采样一个噪声图像作为输入,然后逐步预测噪声图像以生成人脸图像。本发明能够有效提升模型拟合真实图像纹理与细节的能力并提高合成数据多样性。