申请/专利权人：荣耀终端有限公司

申请日：2023-10-18

公开（公告）日：2024-04-09

公开（公告）号：CN117078509B

主分类号：G06T3/04

分类号：G06T3/04;G06N3/0455;G06N3/0464;G06N3/08

优先权：

专利状态码：有效-授权

法律状态：2024.04.09#授权;2023.12.05#实质审查的生效;2023.11.17#公开

摘要：本申请实施例提供一种模型训练方法、照片生成方法及相关设备，所述模型训练方法包括：构建多个图像对，作为预设扩散模型的训练数据，每个图像对包括第一图像和第二图像，第一图像为一视角下图像信息完整的图像，第二图像为相同视角下图像信息缺失的图像；对预设扩散模型进行初始化；基于多个图像对，训练所述预设扩散模型，直至预设扩散模型的损失函数的输出值小于或等于预设阈值，停止训练；将训练后的预设扩散模型确定为照片生成模型。本申请实施例构建包含信息完整图像和信息缺失图像的训练数据对，并对预设扩散模型进行训练，得到照片生成模型，便于通过照片生成模型生成视角调整后的信息完整的照片，使得照片角度符合用户需求。

主权项：1.一种模型训练方法，应用于电子设备，其特征在于，所述方法包括：构建多个图像对，作为预设扩散模型的训练数据，每个图像对包括第一图像和第二图像，所述第一图像为一视角下图像信息完整的图像，所述第二图像为相同视角下图像信息缺失的图像；对所述预设扩散模型进行初始化；基于所述多个图像对，训练所述预设扩散模型，将所述多个图像对中的一图像对输入所述预设扩散模型，计算所述预设扩散模型的损失函数，直至所述预设扩散模型的损失函数的输出值小于或等于预设阈值，停止训练；所述预设扩散模型包括输入层、扩散模块、条件控制模块及输出层，所述扩散模块为稳定扩散模型，包括多个第一编码器块、第一中间块、多个解码器块，所述条件控制模块为控制模型，包括多个零卷积层、多个第二编码器块和第二中间块；所述计算所述预设扩散模型的损失函数，包括：将所述图像对中的所述第一图像和所述第二图像输入所述预设扩散模型的输入层，通过所述输入层对所述第一图像进行预处理；所述输入层将预处理后的所述第一图像输入所述扩散模块，通过所述扩散模块的所述多个第一编码器块分别将所述第一图像编码为潜在图像，并将随机噪声添加至所述潜在图像，并将添加有随机噪声的所述潜在图像传递至所述第一中间块；所述输入层将所述第二图像输入所述条件控制模块，通过所述条件控制模块的所述多个第二编码器块提取所述第二图像的点云特征，并将所述第二图像的点云特征传递至所述第二中间块；所述第二中间块将所述第二图像的点云特征传递至所述第一中间块，所述第一中间块将添加有随机噪声的所述潜在图像和所述第二图像的点云特征传递至所述扩散模块的所述解码器块；所述解码器块根据所述第二图像的点云特征确定所述潜在图像中的预测噪声，从所述潜在图像中减去所述预测噪声，得到更新的潜在图像，并对所述更新的潜在图像进行上采样处理，得到所述第二图像的重建图像；将所述第一图像的图像特征和所述重建图像的图像特征输入所述预设扩散模型的损失函数，得到所述预设扩散模型的损失函数的输出值；将训练后的所述预设扩散模型确定为照片生成模型。

全文数据：

权利要求：

百度查询： 荣耀终端有限公司 模型训练方法、照片生成方法及相关设备

免责声明
1、本报告根据公开、合法渠道获得相关数据和信息，力求客观、公正，但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解，仅供参考使用，不能作为本公司承担任何法律责任的依据或者凭证。

阅读全文 双屏查看 官方信息 专利公告 收藏专利 下载PDF 下载WORD