[發(fā)明專利]圖像增強(qiáng)模型的訓(xùn)練方法及裝置、圖像增強(qiáng)方法及裝置

有效

• 申請(qǐng)?zhí)枺?/td>: 202011475833.5
• 申請(qǐng)日: 2020-12-14
• 公開（公告）號(hào): CN113538304B
• 公開（公告）日: 2023-08-18
• 發(fā)明（設(shè)計(jì)）人: 劉恩雨;李松南
• 申請(qǐng)（專利權(quán)）人: 騰訊科技（深圳）有限公司
• 主分類號(hào): G06T5/50
• 分類號(hào): G06T5/50;G06T7/90;G06N3/0464;G06N3/092;G06N3/096
• 代理公司: 北京勵(lì)誠知識(shí)產(chǎn)權(quán)代理有限公司 11647
• 代理人: 趙爽

本申請(qǐng)?zhí)峁┝艘环N圖像增強(qiáng)模型的訓(xùn)練方法及裝置、圖像增強(qiáng)方法及裝置，能夠在進(jìn)行顏色增強(qiáng)時(shí)，同時(shí)維持增強(qiáng)圖像的細(xì)節(jié)紋理，讓圖像的局部和整體均有很好的增強(qiáng)效果。圖像增強(qiáng)模型的訓(xùn)練方法，包括：獲取訓(xùn)練圖像集，訓(xùn)練圖像集中的各個(gè)訓(xùn)練樣本包括輸入圖像以及配對(duì)的目標(biāo)圖像，其中，目標(biāo)圖像是由輸入圖像經(jīng)色彩增強(qiáng)處理得到的增強(qiáng)圖像與輸入圖像對(duì)應(yīng)的HDR圖像融合后得到的圖像；根據(jù)訓(xùn)練圖像集對(duì)初始圖像增強(qiáng)模型進(jìn)行模型訓(xùn)練，得到目標(biāo)圖像增強(qiáng)模型。

技術(shù)領(lǐng)域

本申請(qǐng)實(shí)施例涉及圖像處理領(lǐng)域，并且更具體地，涉及一種圖像增強(qiáng)模型的訓(xùn)練方法及裝置、圖像增強(qiáng)方法及裝置。

背景技術(shù)

隨著多媒體技術(shù)的普及，人們?cè)谌粘Ｉ詈蜕a(chǎn)活動(dòng)中對(duì)圖像的使用越來越頻繁。圖像的形式多樣，有自然的圖像，例如照片；也有人工圖像，例如那些在計(jì)算機(jī)上通過數(shù)字技術(shù)創(chuàng)造出來的測(cè)試圖像。

目前，圖像增強(qiáng)可以通過對(duì)圖像的像素點(diǎn)或亮度進(jìn)行處理進(jìn)而實(shí)現(xiàn)顏色增強(qiáng)。但無法處理局部的細(xì)節(jié)，因?yàn)檫^暗和過亮的部分的細(xì)節(jié)在增強(qiáng)后，由于顏色的增強(qiáng)使得細(xì)節(jié)會(huì)更微弱。

發(fā)明內(nèi)容

本申請(qǐng)?zhí)峁┝艘环N圖像增強(qiáng)模型的訓(xùn)練方法及裝置、圖像增強(qiáng)方法及裝置、電子設(shè)備、芯片和計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，模型訓(xùn)練中的目標(biāo)圖像是由輸入圖像經(jīng)色彩增強(qiáng)處理得到的增強(qiáng)圖像與輸入圖像對(duì)應(yīng)的HDR圖像融合后得到的圖像，從而目標(biāo)圖像可以將HDR圖像中的細(xì)節(jié)與增強(qiáng)圖像中的色彩和對(duì)比度優(yōu)勢(shì)共同保留，也即，在圖像增強(qiáng)中，能夠在進(jìn)行顏色增強(qiáng)時(shí)，同時(shí)維持增強(qiáng)圖像的細(xì)節(jié)紋理，讓圖像的局部和整體均有很好的增強(qiáng)效果。

本申請(qǐng)的其他特性和優(yōu)點(diǎn)將通過下面的詳細(xì)描述變得顯然，或部分地通過本申請(qǐng)的實(shí)踐而習(xí)得。

根據(jù)本申請(qǐng)的一方面，提供了一種圖像增強(qiáng)模型的訓(xùn)練方法，包括：

獲取訓(xùn)練圖像集，訓(xùn)練圖像集中的各個(gè)訓(xùn)練樣本包括輸入圖像以及配對(duì)的目標(biāo)圖像，其中，目標(biāo)圖像是由輸入圖像經(jīng)色彩增強(qiáng)處理得到的增強(qiáng)圖像與輸入圖像對(duì)應(yīng)的HDR圖像融合后得到的圖像；

根據(jù)訓(xùn)練圖像集對(duì)初始圖像增強(qiáng)模型進(jìn)行模型訓(xùn)練，得到目標(biāo)圖像增強(qiáng)模型。

根據(jù)本申請(qǐng)的一方面，提供了一種圖像增強(qiáng)方法，包括：

獲取待處理圖像；

將待處理圖像輸入目標(biāo)圖像增強(qiáng)模型中，其中，目標(biāo)圖像增強(qiáng)模型通過訓(xùn)練圖像集對(duì)初始圖像增強(qiáng)模型進(jìn)行模型訓(xùn)練得到，訓(xùn)練圖像集中的各個(gè)訓(xùn)練樣本包括輸入圖像以及配對(duì)的目標(biāo)圖像，目標(biāo)圖像是由輸入圖像經(jīng)色彩增強(qiáng)處理得到的增強(qiáng)圖像與輸入圖像對(duì)應(yīng)的HDR圖像融合后得到的圖像；

獲取目標(biāo)圖像增強(qiáng)模型輸出的目標(biāo)增強(qiáng)圖像。

根據(jù)本申請(qǐng)的一方面，提供了一種圖像增強(qiáng)模型的訓(xùn)練裝置，包括：

訓(xùn)練圖像集獲取模塊，用于獲取訓(xùn)練圖像集，訓(xùn)練圖像集中的各個(gè)訓(xùn)練樣本包括輸入圖像以及配對(duì)的目標(biāo)圖像，其中，目標(biāo)圖像是由輸入圖像經(jīng)色彩增強(qiáng)處理得到的增強(qiáng)圖像與輸入圖像對(duì)應(yīng)的HDR圖像融合后得到的圖像；

訓(xùn)練模塊，用于根據(jù)訓(xùn)練圖像集對(duì)初始圖像增強(qiáng)模型進(jìn)行模型訓(xùn)練，得到目標(biāo)圖像增強(qiáng)模型。

根據(jù)本申請(qǐng)的一方面，提供了一種圖像增強(qiáng)裝置，包括：

待處理圖像獲取模塊，用于獲取待處理圖像；

輸入模塊，用于將待處理圖像輸入目標(biāo)圖像增強(qiáng)模型中，其中，目標(biāo)圖像增強(qiáng)模型通過訓(xùn)練圖像集對(duì)初始圖像增強(qiáng)模型進(jìn)行模型訓(xùn)練得到，訓(xùn)練圖像集中的各個(gè)訓(xùn)練樣本包括輸入圖像以及配對(duì)的目標(biāo)圖像，目標(biāo)圖像是由輸入圖像經(jīng)色彩增強(qiáng)處理得到的增強(qiáng)圖像與輸入圖像對(duì)應(yīng)的HDR圖像融合后得到的圖像；

目標(biāo)增強(qiáng)圖像獲取模塊，用于獲取目標(biāo)圖像增強(qiáng)模型輸出的目標(biāo)增強(qiáng)圖像。