[發明專利]圖像著色方法和裝置

有效

• 申請號: 201810123237.7
• 申請日: 2018-02-07
• 公開（公告）號: CN109754444B
• 公開（公告）日: 2021-10-26
• 發明（設計）人: 杜康寧;鄒祥祥
• 申請（專利權）人: 京東方科技集團股份有限公司
• 主分類號: G06T11/00
• 分類號: G06T11/00
• 代理公司: 北京三高永信知識產權代理有限責任公司 11138
• 代理人: 滕一斌

1.一種圖像著色方法，其特征在于，所述圖像著色方法包括：

通過深度學習技術訓練人臉圖像著色模型，包括：獲取人臉圖像樣本；對所述人臉圖像中的人臉進行關鍵點定位；按照所述關鍵點的位置對所述人臉圖像進行尺寸標準化，使得在標準化后的所述人臉圖像中所述關鍵點位于設定位置，所述人臉圖像樣本中的人臉圖像為同一人種的人臉圖像，將所述人臉圖像樣本中的每一張人臉圖像由紅綠藍顏色模型圖像轉換為LAB顏色模型圖像；采用轉換得到的所有的LAB顏色模型圖像對深度卷積生成對抗網絡進行訓練，得到所述人臉圖像著色模型；

通過深度學習技術訓練人物圖像著色模型，包括：獲取人物圖像樣本；對所述人物圖像樣本中的每一張人物圖像進行標準化，其中，所述人物圖像樣本中的人物圖像為同一人種的人物圖像；將所述人物圖像樣本中的每一張人物圖像由紅綠藍顏色模型圖像轉換為LAB顏色模型圖像；采用轉換得到的所有的LAB顏色模型圖像對深度卷積生成對抗網絡進行訓練，得到初次訓練模型；選擇所述人物圖像樣本中的第一人物圖像，組成再次訓練樣本，所述第一人物圖像中第一像素的比例處于第一范圍內，所述第一像素為設定顏色分量的值處于第二范圍內的像素；采用所述再次訓練樣本，對所述初次訓練模型再次訓練，得到所述人物圖像著色模型；

獲取待著色的灰度圖像；

檢測所述灰度圖像中人臉的數量和大小；

當所述灰度圖像中僅存在一個人臉且所述人臉在所述灰度圖像中所占比例的大小超過設定值時，判斷所述灰度圖像為人臉圖像；

當所述灰度圖像中存在至少兩個人臉，或者所述人臉在所述灰度圖像中所占比例的大小未超過設定值時，判斷所述灰度圖像為人物圖像；

將所述灰度圖像輸入到與所述灰度圖像的類型對應的基于神經網絡的圖像著色模型中，得到所述灰度圖像中各個像素的色彩信息；當所述灰度圖像為人臉圖像時，所述圖像著色模型為人臉圖像著色模型，當所述灰度圖像為人物圖像時，所述圖像著色模型為人物圖像著色模型；

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

2.根據權利要求1所述的圖像著色方法，其特征在于，所述將所述灰度圖像和所述色彩信息合成為彩色圖像，包括：

將所述灰度圖像和所述色彩信息合成為LAB顏色模型圖像；

將所述LAB顏色模型圖像轉換為紅綠藍顏色模型圖像。