一种基于扩散模型的人像图换头方法、装置、介质

本公开实施例公开了一种基于扩散模型的人像图换头方法、装置、介质，该方法如下步骤：获取图像信息，该所述图像信息包括源图像与模板图像；构建特征提取模块并训练得到特征提取器，及训练扩散模型的ControlNet模块，并与扩散模型结合形成换头任务模型；将源图像和模板图像输入到换头任务模型得到目标图像。本发明在提取特征时提取了整个人头的特征，同时空间控制条件中含有头发边缘检测，能够确保生成图像的发型相似度。