[发明专利]图像重建模型的训练方法、装置和电子设备

说明书

本公开实施例公开了一种图像重建模型的训练方法、装置和电子设备。该方法的一具体实施方式包括：获取初始全景训练样本图像，根据所述初始全景训练样本图像确定至少一个增强全景训练样本图像，得到全景图像训练样本集，所述初始全景训练样本图像包括至少一个对象图像；从初始全景训练样本图像提取出初始对象样本训练图像，得到对象样本图像训练样本集；基于预设损失函数，利用所述全景图像训练样本集以及所述对象样本图像训练样本集对第一图像处理模型以及第二图像处理模型进行训练。可以一并对第一图像处理模型和第二图像处理模型进行训练，基于收敛后的目标图像处理模型的输出，可以得到更加真实的重建图像。

技术领域

本公开涉及互联网技术领域，尤其涉及一种图像重建模型的训练方法、装置和电子设备。

背景技术

图像重建技术是一种通过物体外部测量的数据，经数字处理获得三维物体的形状信息的技术，已被广泛地应用于医学成像、工业无损检测等各种领域。在对图像重建模型的训练过程中，常常需要确定出对应的损失函数，以利于重建出贴近于真实物体的图像。

发明内容

提供该公开内容部分以便以简要的形式介绍构思，这些构思将在后面的具体实施方式部分被详细描述。该公开内容部分并不旨在标识要求保护的技术方案的关键特征或必要特征，也不旨在用于限制所要求的保护的技术方案的范围。

本公开实施例提供了一种图像重建模型的训练方法、装置和电子设备。

第一方面，本公开实施例提供了一种图像重建模型的训练方法，该方法包括：获取初始全景训练样本图像，根据所述初始全景训练样本图像确定至少一个增强全景训练样本图像，得到全景图像训练样本集，所述初始全景训练样本图像包括至少一个对象图像；从初始全景训练样本图像提取出初始对象样本训练图像，得到对象样本图像训练样本集；基于预设损失函数，利用所述全景图像训练样本集以及所述对象样本图像训练样本集对第一图像处理模型以及第二图像处理模型进行训练，得到训练后的第一目标图像处理模型和第二目标图像处理模型。

第二方面，本公开实施例提供了一种图像重建模型的训练装置，该装置包括：确定模块，用于获取初始全景训练样本图像，根据所述初始全景训练样本图像确定至少一个增强全景训练样本图像，得到全景图像训练样本集，所述初始全景训练样本图像包括至少一个对象图像；提取模块，用于从初始全景训练样本图像提取出初始对象样本训练图像，得到对象样本图像训练样本集；训练模块，用于基于预设损失函数，利用所述全景图像训练样本集以及所述对象样本图像训练样本集对第一图像处理模型以及第二图像处理模型进行训练，得到训练后的第一目标图像处理模型和第二目标图像处理模型。

第三方面，本公开实施例提供了一种电子设备，包括：一个或多个处理器；存储装置，其上存储有一个或多个程序，当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现第一方面所述的图像重建模型的训练方法。

第四方面，本公开实施例提供了一种计算机可读介质，其上存储有计算机程序，该程序被处理器执行时实现上述第一方面所述的图像重建模型的训练方法的步骤。

本公开实施例提供的图像重建模型的训练方法、装置和电子设备，通过获取初始全景训练样本图像，根据所述初始全景训练样本图像确定至少一个增强全景训练样本图像，得到全景图像训练样本集，所述初始全景训练样本图像包括至少一个对象图像；从初始全景训练样本图像提取出初始对象样本训练图像，得到对象样本图像训练样本集；基于预设损失函数，利用所述全景图像训练样本集以及所述对象样本图像训练样本集对第一图像处理模型以及第二图像处理模型进行训练，得到训练后的第一目标图像处理模型和第二目标图像处理模型。可以一并对第一图像处理模型和第二图像处理模型进行训练，基于收敛后的目标图像处理模型的输出，可以得到更加真实的重建图像。

附图说明