图像修复模型的训练方法及图像修复方法、装置及设备

摘要

本申请公开了一种图像修复模型的训练方法、装置及设备，对原图图像进行随机掩码操作，生成缺陷图像。将缺陷图像分离为多个单通道图像，并将多个单通道图像分别输入单通道图像修复模块中，获得每个单通道图像分别对应的单通道预测图像。图像修复模型包括单通道图像修复模块。将多个单通道图像分别对应的单通道预测图像进行合并，获得预测图像。由原图图像和预测图像获得的图像修复模型的损失值训练图像修复模型，直到达到预设条件。基于多个单通道图像的分层修复，能够充分学习图像的语义信息。预测图像的整体像素数值，能够降低已修复区域存在波纹、扭曲等失真情况的发生。如此，使得训练完成的图像修复模型对缺陷图像的修复效果更好。