一种基于扩散模型的脸部水印擦除方法

本发明公开了一种基于扩散模型的脸部水印擦除方法，包括：获取无水印人脸图像、水印数据和水印掩码数据集，进行随机采样生成原始人脸图像和原始水印掩码训练集；结合VAE编码器、DDPM扩散模型和VAE解码器构建学习模型，输入训练集得到原始去水印扩散模型，输入数据集得到验证图像集；将验证图像集难易度分类形成难易样本集，更新数据集采样范围生成新的样本集；基于ResNet18构建双分支分类器，输入难易样本集得到双分支水印数据分类器；输入样本集进行难易度二分类，生成人脸图像训练集和水印掩码训练集；将训练集输入原始去水印扩散模型得到脸部水印擦除扩散模型；根据带水印的人脸图像生成对应的水印分割掩码，输入脸部水印擦除扩散模型，得到无水印的脸部图像；本发明能够针对被水印覆盖的人脸区域进行精确的修复生成，提高生成的去水印图像的效果。