[發明專利]光照估計方法，終端及存儲介質

在審

• 申請號: 202110743512.7
• 申請日: 2021-07-01
• 公開（公告）號: CN113570514A
• 公開（公告）日: 2021-10-29
• 發明（設計）人: 王磊;崔秀芬;凌霄
• 申請（專利權）人: OPPO廣東移動通信有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06N3/04;G06N3/08
• 代理公司: 北京派特恩知識產權代理有限公司 11270
• 代理人: 馬麗;張穎玲

本申請實施例公開了一種光照估計方法，終端及存儲介質，終端利用第一訓練數據訓練初始自編碼模型和初始解碼模型，獲得第一訓練數據對應的第一隱向量和第二隱向量、第一隱向量對應的第一全景圖以及第二隱向量對應的第二全景圖；利用第一隱向量、第二隱向量、第一全景圖以及第二全景圖對初始自編碼模型和初始解碼模型進行優化處理，獲得目標自編碼模型和目標解碼模型；利用第二訓練數據、第一隱向量、第二隱向量以及第一全景圖訓練初始預測模型，獲得目標預測模型；根據目標預測模型和目標解碼模型對輸入圖像進行光照估計處理，獲得輸入圖像對應的天空光全景圖。

技術領域

本發明涉及圖像處理技術領域，尤其涉及一種光照估計方法，終端及存儲介質。

背景技術

室外場景的光照信息會根據天氣或在一天中的時間段而產生較大的差異，例如，中午是明亮的高光和暗影，黃昏時則是溫暖的橙色調。目前，實現室外圖像的光照估計的方法主要包括借助輔助標志物的方法、借助輔助拍攝設備的方法以及無需輔助標志或拍攝設備的圖像分析方法，然而，這類采用參數化光照模型的室外光照預測方法具有很多局限性，比如只能處理晴天的情況，或者其輸出的室外光照參數無法覆蓋全部的天空光形態，光照估計的效果較差。

發明內容

本申請實施例提供了一種光照估計方法，終端及存儲介質，能夠處理任意天氣或時間段下室外場景的天空光形態，具有更優的光照估計能力。

本申請實施例的技術方案是這樣實現的：

第一方面，本申請實施例提供了一種光照估計方法，所述方法包括：

利用第一訓練數據訓練初始自編碼模型和初始解碼模型，獲得所述第一訓練數據對應的第一隱向量和第二隱向量、所述第一隱向量對應的第一全景圖以及所述第二隱向量對應的第二全景圖；

利用所述第一隱向量、所述第二隱向量、所述第一全景圖以及所述第二全景圖對所述初始自編碼模型和初始解碼模型進行優化處理，獲得目標自編碼模型和目標解碼模型；

利用第二訓練數據、所述第一隱向量、所述第二隱向量以及所述第一全景圖訓練初始預測模型，獲得目標預測模型；

根據所述目標預測模型和所述目標解碼模型對所述輸入圖像進行光照估計處理，獲得所述輸入圖像對應的天空光全景圖。

第二方面，本申請實施例提供了一種終端，所述終端包括訓練單元、處理單元以及估計單元，

所述訓練單元，用于利用第一訓練數據訓練初始自編碼模型和初始解碼模型，獲得所述第一訓練數據對應的第一隱向量和第二隱向量、所述第一隱向量對應的第一全景圖以及所述第二隱向量對應的第二全景圖；以及利用第二訓練數據、所述第一隱向量、所述第二隱向量以及所述第一全景圖訓練初始預測模型，獲得目標預測模型；其中，所述目標預測模型用于獲取輸入圖像對應的所述隱向量

所述處理單元，用于利用所述第一隱向量、所述第二隱向量、所述第一全景圖以及所述第二全景圖對所述初始自編碼模型和初始解碼模型進行優化處理，獲得目標自編碼模型和目標解碼模型；

所述估計單元，用于根據所述目標預測模型和所述目標解碼模型對所述輸入圖像進行光照估計處理，獲得所述輸入圖像對應的天空光全景圖。

第三方面，本申請實施例提供了一種終端，所述終端還包括處理器、存儲有所述處理器可執行指令的存儲器，當所述指令被所述處理器執行時，實現如上所述的光照估計方法。

第四方面，本申請實施例提供了一種一種計算機可讀存儲介質，其上存儲有程序，應用于終端中，所述程序被處理器執行時，實現如上所述的光照估計方法。