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

在審

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

本發明提出了一種自動前景提取的數據采集方法及裝置，包括：對待處理的視頻圖像進行前景提取，得到前景圖像；對所述前景圖像以及對應的掩模成對作為一個前景sku信息保存到sku數據庫中，供后續訓練使用；根據訓練任務查找待訓練列表的sku信息并從所述sku數據庫中讀取所需的前景sku信息，并從sku數據庫中拉取樣本作為負樣本；從基礎數據庫中選取背景信息，然后將所述前景sku信息和背景信息進行合成；利用所述負樣本對合成后的數據進行訓練。本發明不需要人工輔助前景提取，做到完全自動化；可以融合各種復雜背景，數據增廣價值巨大。

技術領域

本發明涉及圖像數據處技術領域，特別涉及一種自動前景提取的數據采集方法及裝置。

背景技術

現有的數據集采集包括兩種常見任務，一是檢測任務，二是語義分割任務。其中，最常見的標注方法為人工標注，費時費力，且準確率低。現有的解決方案是利用人工智能輔助的標注方式，例如基于追蹤算法的標注方式，但是這種方式存在以下缺陷：

1、需要事先訓練對應類別的檢測模型；

2、在自動標注的同時會加入背景信息，造成比較單一的背景，從而對數據的多樣性產生影響。

發明內容

本發明的目的旨在至少解決所述技術缺陷之一。

為此，本發明的目的在于提出一種自動前景提取的數據采集方法及裝置。

為了實現上述目的，本發明的實施例提供一種自動前景提取的數據采集方法，包括如下步驟：

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

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

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

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

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

進一步，在所述步驟S1中，采用前景提取算法，對待處理的視頻圖像進行前景提取。

進一步，在所述步驟S1和步驟S2之間，還包括如下步驟：對所述前景圖像進行優化處理，包括平滑處理和填補處理。

進一步，在所述步驟S3中，所述從sku數據庫中拉取樣本作為負樣本，包括如下步驟：當所述訓練任務為非閉合訓練時，從已有的sku信息中計算與該sku信息屬于同一大類但并非同一小類的sku信息，以及與該sku信息不相似的sku信息，從sku數據庫中拉取符合上述條件的樣本作為負樣本。

進一步，在所述步驟S4和步驟S5之間，還包括如下步驟：

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

本發明實施例還提供一種自動前景提取的數據采集裝置，包括：數據采集模塊、前景提取模塊、sku信息查找模塊、前后景合成模塊、訓練模塊，其中，所述數據采集模塊用于對待處理的視頻圖像；所述前景提取模塊用于對所述視頻圖像進行前景提取，得到前景圖像，并將所述前景圖像以及對應的掩模成對作為一個前景sku信息保存到sku數據庫中，供后續訓練使用；所述sku信息查找模塊用于根據訓練任務查找待訓練列表的sku信息并從所述sku數據庫中讀取所需的前景sku信息，并從sku數據庫中拉取樣本作為負樣本；所述前后景合成模塊用于從基礎數據庫中選取背景信息，然后將所述前景sku信息和背景信息進行合成；所述訓練模塊用于利用所述負樣本對合成后的數據進行訓練。