[發(fā)明專利]圖像著色方法和裝置

有效

• 申請(qǐng)?zhí)枺?/td>: 201810123237.7
• 申請(qǐng)日: 2018-02-07
• 公開（公告）號(hào): CN109754444B
• 公開（公告）日: 2021-10-26
• 發(fā)明（設(shè)計(jì)）人: 杜康寧;鄒祥祥
• 申請(qǐng)（專利權(quán)）人: 京東方科技集團(tuán)股份有限公司
• 主分類號(hào): G06T11/00
• 分類號(hào): G06T11/00
• 代理公司: 北京三高永信知識(shí)產(chǎn)權(quán)代理有限責(zé)任公司 11138
• 代理人: 滕一斌

本發(fā)明公開了一種圖像著色方法及裝置，屬于圖像處理技術(shù)領(lǐng)域。所述圖像著色方法包括：獲取待著色的灰度圖像；對(duì)所述灰度圖像進(jìn)行分類，確定所述灰度圖像的類型為人臉圖像或者人物圖像；將所述灰度圖像輸入到與所述灰度圖像的類型對(duì)應(yīng)的基于神經(jīng)網(wǎng)絡(luò)的圖像著色模型中，得到所述灰度圖像中各個(gè)像素的色彩信息；當(dāng)所述灰度圖像為人臉圖像時(shí)，所述圖像著色模型為人臉圖像著色模型，當(dāng)所述灰度圖像為人物圖像時(shí)，所述圖像著色模型為人物圖像著色模型；將所述灰度圖像和所述色彩信息合成為彩色圖像。由于不同類型的灰度圖像的整體顏色以及輪廓差別很大，所以采用不同的著色模型進(jìn)行著色，著色得到的彩色圖像更貼近真實(shí)圖像，效果好。

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

本發(fā)明涉及圖像處理技術(shù)領(lǐng)域，特別涉及一種圖像著色方法和裝置。

背景技術(shù)

隨著移動(dòng)互聯(lián)網(wǎng)的發(fā)展，人們對(duì)圖像視覺效果的要求也越來越高。例如，日常生活中用戶要求照片更加清晰，色彩更加真實(shí)。然而由于技術(shù)原因，年代較為久遠(yuǎn)的照片大多為黑白照片，黑白照片的視覺效果很難與彩色照片相比，所以需要將黑白照片轉(zhuǎn)換為彩色照片。

隨著深度學(xué)習(xí)技術(shù)的發(fā)展，采用神經(jīng)網(wǎng)絡(luò)模型進(jìn)行黑白照片的著色，成為將黑白照片轉(zhuǎn)換為彩色照片的有效手段。然而，目前采用神經(jīng)網(wǎng)絡(luò)模型進(jìn)行著色得到的彩色照片的效果與真實(shí)照片相差較大，效果比較差，同時(shí)處理速度較慢。

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

為了解決現(xiàn)有技術(shù)中采用神經(jīng)網(wǎng)絡(luò)模型進(jìn)行黑白照片的著色，得到的彩色照片的效果與真實(shí)照片相差較大，效果比較差的問題，本發(fā)明實(shí)施例提供了一種圖像著色方法和裝置。所述技術(shù)方案如下：

第一方面，本發(fā)明實(shí)施例提供了一種圖像著色方法，所述圖像著色方法包括：

獲取待著色的灰度圖像；

對(duì)所述灰度圖像進(jìn)行分類，確定所述灰度圖像的類型為人臉圖像或者人物圖像；

將所述灰度圖像輸入到與所述灰度圖像的類型對(duì)應(yīng)的基于神經(jīng)網(wǎng)絡(luò)的圖像著色模型中，得到所述灰度圖像中各個(gè)像素的色彩信息；當(dāng)所述灰度圖像為人臉圖像時(shí)，所述圖像著色模型為人臉圖像著色模型，當(dāng)所述灰度圖像為人物圖像時(shí)，所述圖像著色模型為人物圖像著色模型；

將所述灰度圖像和所述色彩信息合成為彩色圖像。

在本發(fā)明實(shí)施例的一種實(shí)現(xiàn)方式中，所述對(duì)所述灰度圖像進(jìn)行分類，確定所述灰度圖像的類型為人臉圖像或者人物圖像，包括：

檢測(cè)所述灰度圖像中人臉的數(shù)量和大小；

當(dāng)所述灰度圖像中僅存在一個(gè)人臉且所述人臉在所述灰度圖像中所占比例的大小超過設(shè)定值時(shí)，判斷所述灰度圖像為人臉圖像；

當(dāng)所述灰度圖像中存在至少兩個(gè)人臉，或者所述人臉在所述灰度圖像中所占比例的大小未超過設(shè)定值時(shí)，判斷所述灰度圖像為人物圖像。

在本發(fā)明實(shí)施例的另一種實(shí)現(xiàn)方式中，所述將所述灰度圖像和所述色彩信息合成為彩色圖像，包括：

將所述灰度圖像和所述色彩信息合成為L(zhǎng)AB顏色模型圖像；

將所述LAB顏色模型圖像轉(zhuǎn)換為紅綠藍(lán)顏色模型圖像。

在本發(fā)明實(shí)施例的另一種實(shí)現(xiàn)方式中，所述圖像著色方法還包括：

通過深度學(xué)習(xí)技術(shù)訓(xùn)練所述人臉圖像著色模型和所述人物圖像著色模型。

在本發(fā)明實(shí)施例的另一種實(shí)現(xiàn)方式中，所述通過深度學(xué)習(xí)技術(shù)訓(xùn)練所述人臉圖像著色模型，包括：

獲取人臉圖像樣本；

將所述人臉圖像樣本中的每一張人臉圖像由紅綠藍(lán)顏色模型圖像轉(zhuǎn)換為L(zhǎng)AB顏色模型圖像；

采用轉(zhuǎn)換得到的所有的LAB顏色模型圖像對(duì)深度卷積生成對(duì)抗網(wǎng)絡(luò)進(jìn)行訓(xùn)練，得到所述人臉圖像著色模型。