[發明專利]圖像染發處理方法和裝置

有效

• 申請號: 201810463453.6
• 申請日: 2018-05-15
• 公開（公告）號: CN108629819B
• 公開（公告）日: 2019-09-13
• 發明（設計）人: 張惜今
• 申請（專利權）人: 北京字節跳動網絡技術有限公司
• 主分類號: G06T11/40
• 分類號: G06T11/40;G06T7/90
• 代理公司: 北京清亦華知識產權代理事務所(普通合伙) 11201
• 代理人: 張潤

1.一種圖像染發處理方法，其特征在于，包括：

對原始圖像進行計算，獲取反映所述原始圖像中像素透明度的數據矩陣；

獲取所述原始圖像中每個像素的亮度分量，根據所述數據矩陣確定頭發區域，根據所述原始圖像中每個像素的亮度分量計算獲取所述頭發區域的反射系數，根據所述原始圖像中每個像素的亮度分量、所述反射系數和所述數據矩陣確定所述原始圖像中每個像素的光照強度；

根據指定渲染的目標顏色、染色強度以及所述數據矩陣，確定所述原始圖像中每個像素的目標染色值；

根據所述原始圖像中每個像素的光照強度和對應的目標染色值，計算獲取所述原始圖像中每個像素的輸出顏色；

根據所述每個像素的輸出顏色對所述原始圖像進行顏色編輯處理輸出目標圖像。

2.如權利要求1所述的方法，其特征在于，所述對原始圖像進行計算，獲取反映所述原始圖像中像素透明度的數據矩陣，包括：

對所述原始圖像的尺寸進行縮放調整，獲取與預設算法模型尺寸匹配的輸入圖像輸入到所述預設算法模型進行計算，獲取與所述輸入圖像對應的數據矩陣。

3.如權利要求1所述的方法，其特征在于，所述獲取所述原始圖像中每個像素的亮度分量，包括：

對所述原始圖像中每個像素進行顏色空間的數據轉換，獲取所述原始圖像中每個像素的亮度分量；

所述根據所述數據矩陣確定頭發區域包括：

將所述數據矩陣中像素透明度大于預設閾值的區域確定為頭發區域。

4.如權利要求3所述的方法，其特征在于，所述對原始圖像進行計算，獲取反映所述原始圖像中像素透明度的數據矩陣，包括：

判斷所述原始圖像尺寸與預設的卷積神經網絡的輸入數據尺寸是否相同；

若獲知尺寸不相同，則將所述原始圖像尺寸縮放到與所述卷積神經網絡的輸入數據尺寸一致的圖像，輸入到所述卷積神經網絡；

采用所述卷積神經網絡對輸入的圖像進行計算，獲取與所述輸入的圖像對應的數據矩陣；

所述將所述數據矩陣中像素透明度大于預設閾值的區域確定為頭發區域，根據所述原始圖像中每個像素的亮度分量，計算所述頭發區域的反射系數，包括：

將與所述輸入的圖像對應的數據矩陣中像素透明度大于預設閾值的區域確定為頭發區域；

對所述輸入的圖像中每個像素的顏色值進行顏色空間的數據轉換，獲取所述輸入的圖像中每個像素的亮度分量；

根據所述輸入的圖像中每個像素的亮度分量，計算所述頭發區域的反射系數；

在所述根據所述原始圖像中每個像素的亮度分量、所述反射系數和所述數據矩陣，獲取所述原始圖像中每個像素的光照強度之前，還包括：

對與所述輸入的圖像對應的數據矩陣進行縮放處理，獲取與所述原始圖像尺寸相同的數據矩陣。

5.如權利要求3所述的方法，其特征在于，所述根據所述原始圖像中每個像素的亮度分量，計算獲取所述頭發區域的反射系數，包括：

根據所述原始圖像中每個像素的亮度分量，計算整個圖像的第一亮度平均值；

根據所述頭發區域中每個像素的亮度分量，計算所述頭發區域的第二亮度平均值；

采用預設算法對所述第一亮度平均值、所述第二亮度平均值和預設系數進行計算，獲取所述頭發區域的反射系數。

6.如權利要求3所述的方法，其特征在于，所述根據所述原始圖像中每個像素的亮度分量、所述反射系數和所述數據矩陣，計算獲取所述原始圖像中每個像素的光照強度，包括：

計算所述原始圖像中每個像素的亮度分量與所述反射系數的比值獲取所述原始圖像中每個像素的第一光照強度；

采用預設算法對所述原始圖像中每個像素的第一光照強度和所述數據矩陣進行計算，獲取所述原始圖像中每個像素的光照強度。