[发明专利]光照估计方法，终端及存储介质

说明书

本申请实施例公开了一种光照估计方法，终端及存储介质，终端利用第一训练数据训练初始自编码模型和初始解码模型，获得第一训练数据对应的第一隐向量和第二隐向量、第一隐向量对应的第一全景图以及第二隐向量对应的第二全景图；利用第一隐向量、第二隐向量、第一全景图以及第二全景图对初始自编码模型和初始解码模型进行优化处理，获得目标自编码模型和目标解码模型；利用第二训练数据、第一隐向量、第二隐向量以及第一全景图训练初始预测模型，获得目标预测模型；根据目标预测模型和目标解码模型对输入图像进行光照估计处理，获得输入图像对应的天空光全景图。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种光照估计方法，终端及存储介质。

背景技术

室外场景的光照信息会根据天气或在一天中的时间段而产生较大的差异，例如，中午是明亮的高光和暗影，黄昏时则是温暖的橙色调。目前，实现室外图像的光照估计的方法主要包括借助辅助标志物的方法、借助辅助拍摄设备的方法以及无需辅助标志或拍摄设备的图像分析方法，然而，这类采用参数化光照模型的室外光照预测方法具有很多局限性，比如只能处理晴天的情况，或者其输出的室外光照参数无法覆盖全部的天空光形态，光照估计的效果较差。

发明内容

本申请实施例提供了一种光照估计方法，终端及存储介质，能够处理任意天气或时间段下室外场景的天空光形态，具有更优的光照估计能力。

本申请实施例的技术方案是这样实现的：

第一方面，本申请实施例提供了一种光照估计方法，所述方法包括：

利用第一训练数据训练初始自编码模型和初始解码模型，获得所述第一训练数据对应的第一隐向量和第二隐向量、所述第一隐向量对应的第一全景图以及所述第二隐向量对应的第二全景图；

利用所述第一隐向量、所述第二隐向量、所述第一全景图以及所述第二全景图对所述初始自编码模型和初始解码模型进行优化处理，获得目标自编码模型和目标解码模型；

利用第二训练数据、所述第一隐向量、所述第二隐向量以及所述第一全景图训练初始预测模型，获得目标预测模型；

根据所述目标预测模型和所述目标解码模型对所述输入图像进行光照估计处理，获得所述输入图像对应的天空光全景图。

第二方面，本申请实施例提供了一种终端，所述终端包括训练单元、处理单元以及估计单元，

所述训练单元，用于利用第一训练数据训练初始自编码模型和初始解码模型，获得所述第一训练数据对应的第一隐向量和第二隐向量、所述第一隐向量对应的第一全景图以及所述第二隐向量对应的第二全景图；以及利用第二训练数据、所述第一隐向量、所述第二隐向量以及所述第一全景图训练初始预测模型，获得目标预测模型；其中，所述目标预测模型用于获取输入图像对应的所述隐向量

所述处理单元，用于利用所述第一隐向量、所述第二隐向量、所述第一全景图以及所述第二全景图对所述初始自编码模型和初始解码模型进行优化处理，获得目标自编码模型和目标解码模型；

所述估计单元，用于根据所述目标预测模型和所述目标解码模型对所述输入图像进行光照估计处理，获得所述输入图像对应的天空光全景图。

第三方面，本申请实施例提供了一种终端，所述终端还包括处理器、存储有所述处理器可执行指令的存储器，当所述指令被所述处理器执行时，实现如上所述的光照估计方法。

第四方面，本申请实施例提供了一种一种计算机可读存储介质，其上存储有程序，应用于终端中，所述程序被处理器执行时，实现如上所述的光照估计方法。