[發(fā)明專利]光照估計(jì)方法，終端及存儲(chǔ)介質(zhì)

在審

• 申請(qǐng)?zhí)枺?/td>: 202110743512.7
• 申請(qǐng)日: 2021-07-01
• 公開（公告）號(hào): CN113570514A
• 公開（公告）日: 2021-10-29
• 發(fā)明（設(shè)計(jì)）人: 王磊;崔秀芬;凌霄
• 申請(qǐng)（專利權(quán)）人: OPPO廣東移動(dòng)通信有限公司
• 主分類號(hào): G06T5/00
• 分類號(hào): G06T5/00;G06N3/04;G06N3/08
• 代理公司: 北京派特恩知識(shí)產(chǎn)權(quán)代理有限公司 11270
• 代理人: 馬麗;張穎玲

1.一種光照估計(jì)方法，其特征在于，所述方法包括：

利用第一訓(xùn)練數(shù)據(jù)訓(xùn)練初始自編碼模型和初始解碼模型，獲得所述第一訓(xùn)練數(shù)據(jù)對(duì)應(yīng)的第一隱向量和第二隱向量、所述第一隱向量對(duì)應(yīng)的第一全景圖以及所述第二隱向量對(duì)應(yīng)的第二全景圖；

利用所述第一隱向量、所述第二隱向量、所述第一全景圖以及所述第二全景圖對(duì)所述初始自編碼模型和初始解碼模型進(jìn)行優(yōu)化處理，獲得目標(biāo)自編碼模型和目標(biāo)解碼模型；

利用第二訓(xùn)練數(shù)據(jù)、所述第一隱向量、所述第二隱向量以及所述第一全景圖訓(xùn)練初始預(yù)測(cè)模型，獲得目標(biāo)預(yù)測(cè)模型；

根據(jù)所述目標(biāo)預(yù)測(cè)模型和所述目標(biāo)解碼模型對(duì)所述輸入圖像進(jìn)行光照估計(jì)處理，獲得所述輸入圖像對(duì)應(yīng)的天空光全景圖。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述利用所述第一隱向量、所述第二隱向量、所述第一全景圖以及所述第二全景圖對(duì)所述初始自編碼模型和初始解碼模型進(jìn)行優(yōu)化處理，獲得目標(biāo)自編碼模型和目標(biāo)解碼模型，包括：

根據(jù)所述第一隱向量、所述第二隱向量、所述第一全景圖以及所述第二全景圖計(jì)算所述初始自編碼模型和所述初始解碼模型對(duì)應(yīng)的第一損失函數(shù)；

根據(jù)所述第一損失函數(shù)對(duì)所述初始自編碼模型和所述初始解碼模型進(jìn)行優(yōu)化處理，獲得所述目標(biāo)自編碼模型和目標(biāo)解碼模型。

3.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述根據(jù)所述第一隱向量、所述第二隱向量、所述第一全景圖以及所述第二全景圖計(jì)算所述初始自編碼模型和所述初始解碼模型對(duì)應(yīng)的第一損失函數(shù)，包括：

計(jì)算所述第一隱向量和所述第二隱向量對(duì)應(yīng)的第一范數(shù)損失函數(shù)；

計(jì)算所述第一全景圖、所述第二全景圖以及所述第一訓(xùn)練數(shù)據(jù)對(duì)應(yīng)的第二范數(shù)損失函數(shù)；

根據(jù)所述第一范數(shù)損失函數(shù)和所述第二范數(shù)損失函數(shù)計(jì)算所述第一損失函數(shù)。

4.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述計(jì)算所述第一全景圖、所述第二全景圖以及所述第一訓(xùn)練數(shù)據(jù)對(duì)應(yīng)的第二范數(shù)損失函數(shù)，包括：

計(jì)算所述第一全景圖和所述第一訓(xùn)練數(shù)據(jù)對(duì)應(yīng)的第三范數(shù)損失函數(shù)；

計(jì)算所述第二全景圖和所述第一訓(xùn)練數(shù)據(jù)對(duì)應(yīng)的第四范數(shù)損失函數(shù)；

對(duì)所述第三范數(shù)損失函數(shù)和所述第四范數(shù)損失函數(shù)進(jìn)行加法處理，獲得所述第二范數(shù)損失函數(shù)。

5.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述根據(jù)所述第一范數(shù)損失函數(shù)和所述第二范數(shù)損失函數(shù)計(jì)算所述第一損失函數(shù)，包括：

對(duì)所述第一范數(shù)損失函數(shù)和所述第二范數(shù)損失函數(shù)添加第一權(quán)重值；

根據(jù)所述第一權(quán)重值、所述第一范數(shù)損失函數(shù)以及所述第二范數(shù)損失函數(shù)生成所述第一損失函數(shù)。

6.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述利用第二訓(xùn)練數(shù)據(jù)、所述第一隱向量、所述第二隱向量以及所述第一全景圖訓(xùn)練初始預(yù)測(cè)模型，獲得目標(biāo)預(yù)測(cè)模型，包括：

將所述第二訓(xùn)練數(shù)據(jù)輸入至所述初始預(yù)測(cè)模型，輸出所述第二訓(xùn)練數(shù)據(jù)對(duì)應(yīng)的預(yù)測(cè)向量；

將所述預(yù)測(cè)向量輸入至所述目標(biāo)解碼模型，輸出所述預(yù)測(cè)向量對(duì)應(yīng)的第三全景圖；

根據(jù)所述預(yù)測(cè)向量、所述第一隱向量、所述第一全景圖以及所述第三全景圖計(jì)算所述初始預(yù)測(cè)模型對(duì)應(yīng)的第二損失函數(shù)；

根據(jù)所述第二損失函數(shù)對(duì)所述初始預(yù)測(cè)模型進(jìn)行優(yōu)化處理，獲得所述目標(biāo)預(yù)測(cè)模型。

7.根據(jù)權(quán)利要求6所述的方法，其特征在于，所述根據(jù)所述預(yù)測(cè)向量、所述第一隱向量、所述第一全景圖以及所述第三全景圖計(jì)算所述初始預(yù)測(cè)模型對(duì)應(yīng)的第二損失函數(shù)，包括：

計(jì)算所述預(yù)測(cè)向量和所述第一隱向量對(duì)應(yīng)的第五范數(shù)損失函數(shù)；

計(jì)算所述第一全景圖和所述第三全景圖對(duì)應(yīng)的第六范數(shù)損失函數(shù)；

根據(jù)所述第五范數(shù)損失函數(shù)和所述第六范數(shù)損失函數(shù)計(jì)算所述第二損失函數(shù)。

8.根據(jù)權(quán)利要求7所述的方法，其特征在于，所述根據(jù)所述第五范數(shù)損失函數(shù)和所述第六范數(shù)損失函數(shù)計(jì)算所述第二損失函數(shù)，包括：

對(duì)所述第五范數(shù)損失函數(shù)和所述第六范數(shù)損失函數(shù)添加第二權(quán)重值；

根據(jù)所述第二權(quán)重值、所述第五范數(shù)損失函數(shù)以及所述第六范數(shù)損失函數(shù)生成所述第二損失函數(shù)。