[发明专利]图像修复模型的训练方法、装置、计算机设备及存储介质

权利要求书

1.一种图像修复方法，其特征在于，所述方法包括：

获取样本图像对应的掩膜图像，所述掩膜图像用于指示所述样本图像中的显著性物体所在区域；

提取所述样本图像的边缘图像作为第一边缘图像，所述边缘图像用于表征所述样本图像的图像内容的纹理；

基于预先训练的边缘补全模型，对所述第一边缘图像进行边缘补全，得到补全后的边缘图像作为第二边缘图像；

基于所述掩膜图像，获取所述样本图像对应的修复掩膜图像以及上下文掩膜图像，所述修复掩膜图像用于指示所述样本图像中待修复的区域，所述上下文掩膜图像用于指示对待修复的区域进行修复时的已知图像区域；

基于所述样本图像、所述修复掩膜图像、所述上下文掩膜图像以及所述第二边缘图像，对初始模型进行训练，得到训练后的图像修复模型。

2.根据权利要求1所述的方法，其特征在于，所述基于所述样本图像、所述修复掩膜图像、所述上下文掩膜图像以及所述第二边缘图像，对初始模型进行训练，得到训练后的图像修复模型，得到训练后的图像修复模型，包括：

基于所述修复掩膜图像以及上下文掩膜图像，得到背景区域图像，所述背景区域图像用于指示所述待修复的区域以及所述已知图像区域；

基于所述样本图像、所述修复掩膜图像、所述上下文掩膜图像、所述背景区域图像以及所述第二边缘图像，对初始模型进行训练，得到训练后的图像修复模型。

3.根据权利要求2所述的方法，其特征在于，所述基于所述样本图像、所述修复掩膜图像、所述上下文掩膜图像、所述背景区域图像以及所述第二边缘图像，对初始模型进行训练，得到训练后的图像修复模型，包括：

将所述背景区域图像与所述样本图像相乘，得到第一图像；

将所述第一图像与所述修复掩膜图像相加，得到第二图像；

将所述第二图像以及所述第二边缘图像进行图像连接，得到输入图像；

将所述输入图像输入至初始模型，得到所述初始模型输出的输出图像；

基于所述修复掩膜图像以及所述背景区域图像，获取所述输出图像相对所述样本图像的损失值作为总损失值；

基于所述总损失值，对所述初始模型进行迭代训练，直至所述初始模型满足预设条件，得到练后的图像修复模型。

4.根据权利要求3所述的方法，其特征在于，所述基于所述修复掩膜图像以及所述背景区域图像，获取所述输出图像相对所述样本图像的损失值作为总损失值，包括：

从所述样本图像中获取所述修复掩膜图像指示的区域作为第一区域图像，以及从所述输出图像中获取所述修复掩膜图像指示的区域作为第二区域图像；

获取所述第二区域图像相对所述第一区域图像的差异，作为第一损失值；

从所述样本图像中获取所述背景区域图像指示的区域作为第三区域图像，以及从所述输出图像中获取所述背景区域图像指示的区域作为第四区域图像；

获取所述第四区域图像相对所述第三区域图像的差异，作为第二损失值；

基于所述第一损失值以及所述第二损失值，得到所述输出图像相对所述样本图像的损失值作为总损失值。

5.根据权利要求2所述的方法，其特征在于，所述基于所述修复掩膜图像以及上下文掩膜图像，得到背景区域图像，包括：

将所述修复掩膜图像与所述上下文掩膜图像进行相加，得到背景区域图像。

6.根据权利要求1所述的方法，其特征在于，所述基于所述掩膜图像，获取所述样本图像对应的修复掩膜图像以及上下文掩膜图像，包括：

对所述掩膜图像进行膨胀处理，得到第一处理图像；

对所述掩膜图像进行腐蚀处理，得到第二处理图像；

基于所述掩膜图像、所述第一处理图像以及所述第二处理图像，获取所述样本图像对应的修复掩膜图像以及上下文掩膜图像。