一种基于扩散模型的头发绘制方法及系统

本申请涉及一种基于扩散模型的头发绘制方法及系统，方法包括：根据用户在原始图片上录入的涂抹指令确定画发目标区域，并生成画发目标区域mask图；将原始图片输入头发分割模型，得到原图头发mask图；对原始图片中的画发目标区域进行填充，得到mask区域发色填充结果图；根据画发目标区域mask图和mask区域发色填充结果图，得到扩散模型画发控制注入条件；将原始图片和扩散模型画发控制注入条件输入扩散模型，使扩散模型根据扩散模型画发控制注入条件将原始图片中的待画发区域重绘为头发，得到扩散模型画发结果图；以原始图片中原图头发mask图指示的头发区域作为发色参考，对扩散模型画发结果图中画发目标区域mask图指示的头发发色进行矫正；将矫正发色后的画发结果图输出。