[發明專利]一種自動前景提取的數據采集方法及裝置

在審

• 申請號: 201910049969.0
• 申請日: 2019-01-18
• 公開（公告）號: CN109829932A
• 公開（公告）日: 2019-05-31
• 發明（設計）人: 張發恩;趙江華;秦永強
• 申請（專利權）人: 創新奇智（寧波）科技有限公司
• 主分類號: G06T7/194
• 分類號: G06T7/194;G06T5/00
• 代理公司: 北京中企鴻陽知識產權代理事務所(普通合伙) 11487
• 代理人: 李斌

1.一種自動前景提取的數據采集方法，其特征在于，包括如下步驟：

步驟S1，對待處理的視頻圖像進行前景提取，得到前景圖像；

步驟S2，對所述前景圖像以及對應的掩模成對作為一個前景sku信息保存到sku數據庫中，供后續訓練使用；

步驟S3，根據訓練任務查找待訓練列表的sku信息并從所述sku數據庫中讀取所需的前景sku信息，并從sku數據庫中拉取樣本作為負樣本；

步驟S4，從基礎數據庫中選取背景信息，然后將所述前景sku信息和背景信息進行合成；

步驟S5，利用所述負樣本對合成后的數據進行訓練。

2.如權利要求1所述的自動前景提取的數據采集方法，其特征在于，在所述步驟S1中，采用前景提取算法，對待處理的視頻圖像進行前景提取。

3.如權利要求1所述的自動前景提取的數據采集方法，其特征在于，在所述步驟S1和步驟S2之間，還包括如下步驟：對所述前景圖像進行優化處理，包括平滑處理和填補處理。

4.如權利要求1所述的自動前景提取的數據采集方法，其特征在于，在所述步驟S3中，所述從sku數據庫中拉取樣本作為負樣本，包括如下步驟：當所述訓練任務為非閉合訓練時，從已有的sku信息中計算與該sku信息屬于同一大類但并非同一小類的sku信息，以及與該sku信息不相似的sku信息，從sku數據庫中拉取符合上述條件的樣本作為負樣本。

5.如權利要求1所述的自動前景提取的數據采集方法，其特征在于，在所述步驟S4和步驟S5之間，還包括如下步驟：

對步驟S4合成后的數據進行數據增強處理，然后將數據增強處理后的合成數據送入步驟S5進行訓練。

6.一種自動前景提取的數據采集裝置，其特征在于，包括：數據采集模塊、前景提取模塊、sku信息查找模塊、前后景合成模塊、訓練模塊，其中，

所述數據采集模塊用于對待處理的視頻圖像；

所述前景提取模塊用于對所述視頻圖像進行前景提取，得到前景圖像，并將所述前景圖像以及對應的掩模成對作為一個前景sku信息保存到sku數據庫中，供后續訓練使用；

所述sku信息查找模塊用于根據訓練任務查找待訓練列表的sku信息并從所述sku數據庫中讀取所需的前景sku信息，并從sku數據庫中拉取樣本作為負樣本；

所述前后景合成模塊用于從基礎數據庫中選取背景信息，然后將所述前景sku信息和背景信息進行合成；

所述訓練模塊用于利用所述負樣本對合成后的數據進行訓練。

7.如權利要求6所述的自動前景提取的數據采集裝置，其特征在于，所述前景提取模塊用于采用前景提取算法，對待處理的視頻圖像進行前景提取。

8.如權利要求6所述的自動前景提取的數據采集裝置，其特征在于，還包括：掩模后處理模塊，所述掩模后處理模塊分別與所述前景提取模塊相連，用于對所述前景圖像進行優化處理，包括平滑處理和填補處理，然后將優化處理后的前景圖像以及對應的掩模成對作為一個前景sku信息保存到sku數據庫中，供后續訓練使用。

9.如權利要求6所述的自動前景提取的數據采集裝置，其特征在于，當所述訓練任務為非閉合訓練時，所述sku信息查找模塊從已有的sku信息中計算與該sku信息屬于同一大類但并非同一小類的sku信息，以及與該sku信息不相似的sku信息，從sku數據庫中拉取符合上述條件的樣本作為負樣本。

10.如權利要求6所述的自動前景提取的數據采集裝置，其特征在于，還包括：數據增強處理模塊，所述數據增強處理模塊與所述前后景合成模塊和所述訓練模塊連接，用于對所述前后景合成模塊合成后的數據進行數據增強處理，然后將數據增強處理后的合成數據發送至所述訓練模塊，由所述訓練模塊利用所述負樣本對合成后的數據進行訓練。