模型训练方法、图像重建方法、装置、电子设备及介质

本申请公开了一种模型训练方法、图像重建方法、装置、电子设备及介质，涉及神经网络技术领域。方法先将初始训练图像输入至初始的图像重建模型；通过完成预训练的分割网络模块进行图像分割处理，得到分割掩码概率训练图像，将分割掩码概率训练图像与初始训练图像输入至扩散网络模块的加噪单元，得到加噪训练图像；将加噪训练图像输入至扩散网络模块的去噪单元，得到预测训练噪声与重建训练图像；基于重建训练图像、初始训练图像、预测训练噪声、分割掩码概率训练图像、真实标签图像，计算得到总损失，基于总损失对初始的图像重建模型进行迭代更新，得到训练好的目标图像重建模型。能够提高图像重建模型对于具有遮挡区域的图像的重建质量。