[發明專利]一種陳列圖像樣本的生成方法及裝置

在審

• 申請號: 201911284109.1
• 申請日: 2019-12-13
• 公開（公告）號: CN111062861A
• 公開（公告）日: 2020-04-24
• 發明（設計）人: 夏志鵬;丁明;陳永輝;李海榮
• 申請（專利權）人: 廣州市玄武無線科技股份有限公司
• 主分類號: G06T3/00
• 分類號: G06T3/00;G06K9/00
• 代理公司: 廣州三環專利商標代理有限公司 44202
• 代理人: 郭浩輝;麥小嬋

1.一種陳列圖像樣本的生成方法，其特征在于，包括：

獲取不同視覺角度下的待融合SKU的SKU圖像，并根據視覺角度對每一所述SKU圖像進行分類；

獲取陳列場景的場景圖像，并根據視覺角度對所述場景圖像中的每一待替換SKU進行分類；其中，所述場景圖像中的待替換SKU與所述SKU圖像中的SKU為不同品類的SKU；

以所述SKU圖像為前景圖像，以所述場景圖像為背景圖像，根據所述前景圖像所屬的視覺角度類別，將所述前景圖像中的待融合SKU與所述背景圖像中，視覺角度類別相同的待替換SKU進行替換，生成所述陳列圖像樣本。

2.根據權利要求1所述陳列圖像樣本的生成方法，其特征在于，所述以所述SKU圖像為前景圖像，以所述場景圖像為背景圖像，根據所述前景圖像所屬的視覺角度類別，將所述前景圖像中的待融合SKU與所述背景圖像中，視覺角度類別相同的待替換SKU進行替換，生成所述陳列圖像樣本，具體包括：

步驟A:以一所述SKU圖像，作為所述前景圖像，并提取所述前景圖像中待融合SKU的mask；

步驟B:以一所述場景圖像，作為所述背景圖像，并提取所述背景圖像中所有待替換SKU對應的標注框的位置信息和視覺角度類別信息；

步驟C:選取一標注框，并判斷所選取的標注框的視覺角度類別信息是否與所述前景圖像的視覺角度類別一致；若一致，則將所選取的標注框對應的待替換SKU與所述待融合SKU的mask進行替換，若不一致，則重新選取一標注框，重新判斷所選取的標注框的視覺角度類別信息是否與所述前景圖像的視覺角度類別一致；

步驟D:重復步驟A和步驟C直至所替換的前景圖像的數量達到預設值，生成所述陳列圖像樣本。

3.根據權利要求1所述陳列圖像樣本的生成方法，其特征在于，在獲取不同視覺角度下的待融合SKU的SKU圖像之后，還包括：將所有所述SKU圖像輸入預設的對抗網絡模型中，以使所述對抗網絡模型生成新的SKU圖像。

4.根據權利要求1所述陳列圖像樣本的生成方法，其特征在于，在生成所述陳列圖像樣本后，還包括，通過預設的StyleGAN模型對所述陳列圖像樣本進行處理。

5.根據權利要求2所述陳列圖像樣本的生成方法，其特征在于，還包括：修改所述陳列圖像樣本中待融合SKU的mask所對應的標注框的位置信息，繼而計算所述陳列圖像樣本中每兩個標注框的iou值，并根據所述iou值進行去重。

6.根據權利要求1所述陳列圖像樣本的生成方法，其特征在于，所述陳列場景包括：貨架場景、冰柜場景、端架場景或地推場景。

7.一種陳列圖像樣本的生成裝置，其特征在于，包括SKU圖像獲取及處理模塊，場景圖像獲取及處理模塊和圖像融合模塊；

所述SKU圖像獲取及處理模塊，用于獲取不同視覺角度下的待融合SKU的SKU圖像，并根據視覺角度對每一所述SKU圖像進行分類；

所述場景圖像獲取及處理模塊，用于獲取陳列場景的場景圖像，并根據視覺角度對所述場景圖像中的每一待替換SKU進行分類；其中，所述場景圖像中的待替換SKU與所述SKU圖像中的SKU為不同品類的SKU；

所述圖像融合模塊，用于以所述SKU圖像為前景圖像，以所述場景圖像為背景圖像，根據所述前景圖像所屬的視覺角度類別，將所述前景圖像中的待融合SKU與所述背景圖像中，視覺角度類別相同的待替換SKU進行替換，生成所述陳列圖像樣本。