[發明專利]人物線稿生成方法、裝置、設備及介質

在審

• 申請號: 202210912212.1
• 申請日: 2022-07-28
• 公開（公告）號: CN115170388A
• 公開（公告）日: 2022-10-11
• 發明（設計）人: 方承煜;韓先鋒
• 申請（專利權）人: 西南大學
• 主分類號: G06T3/00
• 分類號: G06T3/00;G06N3/04;G06N3/08
• 代理公司: 上海光華專利事務所(普通合伙) 31219
• 代理人: 唐勇

1.一種人物線稿生成方法，其特征在于，所述方法包括：

獲取目標人物圖片；

將所述目標人物圖片輸入至預先訓練好的人物線稿模型中，通過所述人物線稿模型對所述目標人物圖片進行轉換，得到人物線稿；其中，所述人物線稿模型的訓練方法，包括：

獲取人物圖片和與人物圖片對應的人物線稿；

將人物圖片輸入至人物線稿模型的生成器中，生成與人物圖片對應的線條圖，以及將所述線條圖與所述人物圖片對應的人物線稿輸入至人物線稿模型的判別器中；

分別計算第一損失函數、第二損失函數、第三損失函數，所述第一損失函數為生成器的損失函數，所述第二損失函數用于表示所述人物線稿與所述線條圖之間的誤差，所述第三損失函數為判別器的損失函數；

對所述第一損失函數、所述第二損失函數、所述第三損失函數進行加權求和，得到總損失函數；

對所述生成器與所述判別器的參數進行迭代更新，使所述總損失函數達到預設條件，以完成所述人物線稿模型的訓練。

2.根據權利要求1所述的人物線稿生成方法，其特征在于，所述生成器包括：

編碼器，用于對所述人物圖片進行編碼，得到編碼結果；

解碼器，用于對所述編碼結果進行解碼，得到線條圖；

所述編碼器包括多層編碼層，用于對所述人物圖片進行多尺度采樣，并對采樣得到的多尺度特征進行特征融合，得到融合特征；

所述解碼器包括多層解碼層，在對所述編碼結果進行解碼的過程中，將每個編碼層輸出的特征圖嵌入到不同的解碼層中。

3.根據權利要求2所述的人物線稿生成方法，其特征在于，所述編碼器采用預先訓練好的ResNeXt-50作為編碼器，編碼層表示為：

第n-i+1層編碼層的輸入表示為

其中，表示n-i層編碼層的輸出，到表示第1到第i層編碼層的輸出，PDS()表示漸進式降采樣操作，表示求和操作；

UP()表示對特征圖進行上采樣；

表示第i層編碼層的輸出，表示第i層編碼層的輸入，C表示組的數量，表示變換函數。

4.根據權利要求3所述的人物線稿生成方法，其特征在于，所述判別器包括多個依次連接的判別器塊，每一個判別器塊包括依次連接的卷積層、歸一化層和殘差層。

5.根據權利要求1所述的人物線稿生成方法，其特征在于，所述第一損失函數表示為：

其中，f₁為均方誤差損失，f₂表示平均運算，f₄對輸入的圖片進行調整，其大小與判別器D的輸出相同，f₃和f₅分別為標量值1和0的填充張量。

6.根據權利要求5所述的人物線稿生成方法，其特征在于，所述第三損失函數表示為：

7.根據權利要求6所述的人物線稿生成方法，其特征在于，所述第二損失函數表示為：