[發明專利]色調映射模型的訓練方法、色調映射方法及電子設備

有效

• 申請號: 202010562136.7
• 申請日: 2020-06-18
• 公開（公告）號: CN111784598B
• 公開（公告）日: 2023-06-02
• 發明（設計）人: 何慕威
• 申請（專利權）人: OPPO（重慶）智能科技有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06T5/40;G06N3/0464;G06N3/08;G06N20/00
• 代理公司: 北京派特恩知識產權代理有限公司 11270
• 代理人: 馬麗;張穎玲

本申請公開了色調映射模型的訓練方法、色調映射方法、及電子設備、存儲介質。其中，色調映射模型的訓練方法包括：確定至少一組訓練圖像對；所述至少一組訓練圖像對中的每一組訓練圖像對由源訓練圖像和對應的目標訓練圖像構成；構成一組訓練圖像對的源訓練圖像和對應的目標訓練圖像為關于相同圖像內容的兩張亮度不同的低動態范圍LDR圖像；所述目標訓練圖像為滿足設定的圖像質量指標的圖像；采用所述至少一組訓練圖像對訓練色調映射模型；其中，所述色調映射模型用于改變輸入圖像中至少一個像素的像素狀態，所述像素狀態包括：像素的顏色和/或亮度。

技術領域

本申請涉及影像技術領域，尤其涉及色調映射模型的訓練方法、色調映射方法及電子設備、存儲介質。

背景技術

色調映射（Tone?Mapping）是在有限動態范圍媒介上近似顯示高動態范圍（HDR，High-Dynamic?Range）圖像的一項計算機圖形技術。色調映射的方法分為對圖像進行全局統一色調映射處理的算法和對圖像進行局部色調映射處理的算法。

相關技術中，全局統一色調映射處理的算法性能一般劣于局部色調映射處理的方法，而局部色調映射處理的方法會出現光暈等現象。因而，在低動態范圍（LDR，Low-DynamicRange）顯示媒介上如何更好地表達原HDR圖像的亮度信息與明暗對比特征，一直是當前色調映射的首要問題。

發明內容

本申請實施例提供一種色調映射模型的訓練方法、色調映射方法及電子設備、存儲介質，以至少解決相關技術中如何在LDR顯示媒介上更好地表達HDR圖像的亮度信息與明暗對比特征的問題。

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

第一方面，本申請實施例提供了一種色調映射模型的訓練方法，該方法包括：

確定至少一組訓練圖像對；所述至少一組訓練圖像對中的每一組訓練圖像對由源訓練圖像和對應的目標訓練圖像構成；構成一組訓練圖像對的源訓練圖像和對應的目標訓練圖像為關于相同圖像內容的兩張亮度不同的低動態范圍LDR圖像；所述目標訓練圖像為滿足設定的圖像質量指標的圖像；

采用所述至少一組訓練圖像對訓練色調映射模型；其中，所述色調映射模型用于改變輸入圖像中至少一個像素的像素狀態，所述像素狀態包括：像素的顏色和/或亮度。

第二方面，本申請實施例還提供了一種色調映射方法，該方法包括：

對HDR圖像進行第一色調映射，生成第一LDR圖像；

將所述第一LDR圖像輸入訓練完成的色調映射模型，得到第二LDR圖像；其中，

所述色調映射模型采用上述任一項所述的色調映射模型的訓練方法訓練得到。

第三方面，本申請實施例提供了一種電子設備，該電子設備包括：

確定單元，用于確定至少一組訓練圖像對；所述至少一組訓練圖像對中的每一組訓練圖像對由源訓練圖像和對應的目標訓練圖像構成；構成一組訓練圖像對的源訓練圖像和對應的目標訓練圖像為關于相同圖像內容的兩張亮度不同的LDR圖像；所述目標訓練圖像為滿足設定的圖像質量指標的圖像；

訓練單元，用于采用所述至少一組訓練圖像對訓練色調映射模型；其中，所述色調映射模型用于改變輸入圖像中至少一個像素的像素狀態，所述像素狀態包括：像素的顏色和/或亮度。

第四方面，本申請實施例提供了一種電子設備，該電子設備包括：

第一生成單元，用于對HDR圖像進行第一色調映射，生成第一LDR圖像；

第二生成單元，用于將所述第一LDR圖像輸入訓練完成的色調映射模型，得到第二LDR圖像；其中，

所述色調映射模型采用上述任一項所述的色調映射模型的訓練方法訓練得到。