一种基于扩散模型的发型替换视频生成方法及装置

本申请属于图像处理技术领域，公开了一种基于扩散模型的发型替换视频生成方法及装置，该方法包括：获取发型图像数据集并训练扩散模型，得到发型生成模型；获取用户输入的人物图像和目标发型提示词；采用骨骼关键点检测技术识别人物图像，得到骨骼姿态数据；采用图像分割算法划分人物图像中的头发区域，得到头发蒙版；通过高斯模糊算法和羽化算法对头发蒙版进行平滑处理和羽化处理；将头发蒙版和目标发型提示词输入发型生成模型，并将骨骼姿态数据作为发型生成模型的约束，得到发型替换图像；基于人物图像和发型替换图像生成发型替换视频。本申请能够提高发型替换图像和视频的真实性与自然度。