[發明專利]一種神經網絡的訓練方法及圖像匹配方法、裝置

有效

• 申請號: 201811535420.4
• 申請日: 2018-12-14
• 公開（公告）號: CN109670591B
• 公開（公告）日: 2022-09-27
• 發明（設計）人: 葛玉瑩;吳凌云;張瑞茂;羅平
• 申請（專利權）人: 深圳市商湯科技有限公司
• 主分類號: G06N3/08
• 分類號: G06N3/08;G06V10/74;G06V10/44;G06V10/82
• 代理公司: 北京派特恩知識產權代理有限公司 11270
• 代理人: 蔣雅潔;張穎玲

本申請公開一種神經網絡的訓練方法及圖像匹配方法、裝置，至少包括：標注第一服裝實例和第二服裝實例的注釋信息，所述第一服裝實例和第二服裝實例分別來源于第一服裝圖像和第二服裝圖像；響應于所述第一服裝實例和所述第二服裝實例匹配的情況，將所述第一服裝圖像和所述第二服裝圖像進行配對；基于配對的所述第一服裝圖像和所述第二服裝圖像對待訓練的神經網絡進行訓練。

技術領域

本申請涉及服裝圖像解析技術，尤其涉及一種神經網絡的訓練方法及圖像匹配方法、裝置。

背景技術

服裝圖像解析，因其在學術界和工業界的巨大潛力，成為近年來逐漸熱門的研究領域。然而，在實際應用中，服裝理解仍然面臨著諸多挑戰。比如數據方面，服裝數據集(DeepFashion)成為現有的最大服裝數據集，但DeepFashion有其自身的缺陷，比如，每張圖像中只有單件服裝實例的注釋，如此定義的基準數據集與實際情況之間的差距，會嚴重影響服裝理解的應用。

申請內容

為解決上述技術問題，本申請實施例提供了一種神經網絡的訓練方法及圖像匹配方法、裝置、存儲介質、計算機程序產品、計算機設備。

本申請實施例提供的神經網絡的訓練方法，包括：

標注第一服裝實例和第二服裝實例的注釋信息，所述第一服裝實例和第二服裝實例分別來源于第一服裝圖像和第二服裝圖像；

響應于所述第一服裝實例和所述第二服裝實例匹配的情況，將所述第一服裝圖像和所述第二服裝圖像進行配對；

基于配對的所述第一服裝圖像和所述第二服裝圖像對待訓練的神經網絡進行訓練。

本申請實施例中，所述標注第一服裝實例和第二服裝實例的注釋信息，包括：

分別標注所述第一服裝實例和所述第二服裝實例的服裝邊界框。

本申請實施例中，所述標注第一服裝實例和第二服裝實例的注釋信息，還包括：

分別標注所述第一服裝實例和所述第二服裝實例的服裝類別和關鍵點。

本申請實施例中，所述標注第一服裝實例和第二服裝實例的注釋信息，還包括：分別標注所述第一服裝實例和所述第二服裝實例的服裝輪廓線以及分割掩碼注釋。

本申請實施例中，所述分別標注所述第一服裝實例和所述第二服裝實例的服裝類別和關鍵點，包括：

分別獲取所述第一服裝實例和所述第二服裝實例的服裝類別；

基于所述服裝類別的標注規則分別標注出所述第一服裝實例和所述第二服裝實例的對應關鍵點。

本申請實施例中，所述分別標注所述第一服裝實例和所述第二服裝實例的服裝類別和關鍵點之后，還包括：

標注出每個所述關鍵點的屬性信息，所述屬性信息用于表明所述關鍵點是屬于可見點還是屬于遮擋點。

本申請實施例中，所述標注第一服裝實例和第二服裝實例的注釋信息，還包括：

分別標注出所述第一服裝實例和所述第二服裝實例的邊緣點和交界點，其中，所述邊緣點是指所述服裝實例處于服裝圖像邊界上的點，所述交界點是指所述第一服裝實例或者所述第二服裝實例與其他服裝實例相交界的地方用于繪制服裝輪廓線的點。

本申請實施例中，所述分別標注所述第一服裝實例和所述第二服裝實例的服裝輪廓線，包括：

分別基于所述第一服裝實例和第二服裝實例的關鍵點、每個關鍵點的屬性信息、邊緣點和交界點，分別繪制所述第一服裝實例和所述第二服裝實例的服裝輪廓線。

本申請實施例中，所述分別標注所述第一服裝實例和所述第二服裝實例的分割掩碼注釋，包括：