[發(fā)明專利]生成式對抗網(wǎng)絡(luò)的訓(xùn)練方法、圖像處理方法、裝置和設(shè)備

在審

• 申請?zhí)枺?/td>: 202010965427.0
• 申請日: 2020-09-14
• 公開（公告）號: CN112101204A
• 公開（公告）日: 2020-12-18
• 發(fā)明（設(shè)計）人: 何聲一;劉家銘;洪智濱
• 申請（專利權(quán)）人: 北京百度網(wǎng)訊科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06N3/04;G06N3/08
• 代理公司: 中科專利商標(biāo)代理有限責(zé)任公司 11021
• 代理人: 范芳茗

1.一種生成式對抗網(wǎng)絡(luò)的訓(xùn)練方法，其特征在于，包括：

獲取多組圖像集，其中，所述多組圖像集中的每組圖像集具有各自的年齡段信息，所述每組圖像集中的每個圖像包括面部信息，所述面部信息的年齡信息屬于與所述每組圖像集對應(yīng)的年齡段；以及

基于所述多組圖像集和所述多組圖像集各自的年齡段信息，對預(yù)先構(gòu)建的初始網(wǎng)絡(luò)模型進(jìn)行訓(xùn)練，得到經(jīng)訓(xùn)練的所述生成式對抗網(wǎng)絡(luò)。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述初始網(wǎng)絡(luò)模型包括生成網(wǎng)絡(luò)和判別網(wǎng)絡(luò)；

所述基于所述多組圖像集和所述多組圖像集各自的年齡段信息，對預(yù)先構(gòu)建的初始網(wǎng)絡(luò)模型進(jìn)行訓(xùn)練包括：針對所述每組圖像集中的任一圖像，

確定所述任一圖像中的面部信息的第一目標(biāo)年齡段；

利用所述生成網(wǎng)絡(luò)對所述任一圖像進(jìn)行處理，得到生成圖像；

利用所述判別網(wǎng)絡(luò)對所述生成圖像進(jìn)行處理，得到所述生成圖像來自所述多組圖像集的概率值以及與所述生成圖像對應(yīng)的年齡段；

根據(jù)與所述生成圖像對應(yīng)的年齡段和所述第一目標(biāo)年齡段之間的差值以及所述概率值，確定所述生成網(wǎng)絡(luò)的損失值；

根據(jù)所述概率值，確定所述判別網(wǎng)絡(luò)的損失值；

根據(jù)所述生成網(wǎng)絡(luò)的損失值和所述判別網(wǎng)絡(luò)的損失值分別調(diào)整所述生成網(wǎng)絡(luò)的參數(shù)和所述判別網(wǎng)絡(luò)的參數(shù)，以得到更新的初始網(wǎng)絡(luò)模型；以及

針對所述更新的初始網(wǎng)絡(luò)模型，重復(fù)上述訓(xùn)練過程，直至所述生成網(wǎng)絡(luò)輸出的生成圖像的清晰度大于預(yù)設(shè)閾值，得到經(jīng)訓(xùn)練的所述生成式對抗網(wǎng)絡(luò)。

3.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述生成網(wǎng)絡(luò)包括形變處理層、紋理處理層、灰度處理層和融合處理層；

所述利用所述生成網(wǎng)絡(luò)對所述任一圖像進(jìn)行處理，得到生成圖像包括：

利用所述形變處理層對所述任一圖像中的面部信息的至少部分像素點(diǎn)的位置進(jìn)行調(diào)整，得到第一特征圖像；

利用所述紋理處理層對所述第一特征圖像進(jìn)行處理，得到第二特征圖像；

利用所述灰度處理層對所述第一特征圖像進(jìn)行處理，得到第三特征圖像，其中，所述第三特征圖像為灰度圖像；以及

利用所述融合處理層根據(jù)所述第三特征圖像中每個像素點(diǎn)的像素值，對所述第一特征圖像和所述第二特征圖像進(jìn)行融合，得到所述生成圖像。

4.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述利用所述紋理處理層對所述第一特征圖像進(jìn)行處理，得到第二特征圖像包括：

利用殘差網(wǎng)絡(luò)對所述第一特征圖像進(jìn)行處理，得到所述第二特征圖像。

5.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述利用所述灰度處理層對所述第一特征圖像進(jìn)行處理，得到第三特征圖像包括：

利用殘差網(wǎng)絡(luò)對所述第一特征圖像進(jìn)行處理，得到所述第二特征圖像；以及

將所述第二特征圖像轉(zhuǎn)換為灰度圖像，得到所述第三特征圖像。

6.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述利用所述融合處理層根據(jù)所述第三特征圖像中每個像素點(diǎn)的像素值，對所述第一特征圖像和所述第二特征圖像進(jìn)行融合包括：

針對所述第三特征圖像中的任一像素點(diǎn)的像素值為G，

確定與所述任一像素點(diǎn)對應(yīng)的所述第二特征圖像中的像素點(diǎn)的權(quán)重為W_G；

確定與所述任一像素點(diǎn)對應(yīng)的所述第一特征圖像中的像素點(diǎn)的權(quán)重為(1-W_G)；

根據(jù)所述W_G和所述(1-W_G)，對所述第一特征圖像和所述第二特征圖像進(jìn)行融合。

7.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述利用所述判別網(wǎng)絡(luò)對所述生成圖像進(jìn)行處理包括：

獲取所述生成網(wǎng)絡(luò)輸出的多個生成圖像；

基于所述多組圖像集、所述多組圖像集各自的年齡段信息和所述多個生成圖像，對預(yù)先構(gòu)建的初始分類模型進(jìn)行訓(xùn)練，以生成經(jīng)訓(xùn)練的所述判別網(wǎng)絡(luò)；以及

利用所述經(jīng)訓(xùn)練的所述判別網(wǎng)絡(luò)對所述生成圖像進(jìn)行處理，得到所述生成圖像來自所述多組圖像集的概率值以及與所述生成圖像對應(yīng)的年齡段。