[發明專利]目標檢測數據集生成方法、裝置、系統和電子設備

在審

• 申請號: 202010850279.8
• 申請日: 2020-08-21
• 公開（公告）號: CN111951259A
• 公開（公告）日: 2020-11-17
• 發明（設計）人: 熊得竹;江德智;周德成;溫志慶
• 申請（專利權）人: 季華實驗室
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06T7/70;G06T11/60
• 代理公司: 佛山市海融科創知識產權代理事務所(普通合伙) 44377
• 代理人: 陳志超;黃家豪

本發明提供了一種目標檢測數據集生成方法、裝置、系統和電子設備，通過獲取所述采集臺采集的多張不同角度的工件的第一圖片；提取每張所述第一圖片中的工件圖像；把所述工件圖像放入預存的背景圖片中，得到第二圖片；根據工件圖像的位置和大小在所述第二圖片中生成邊框信息和標簽；從而可快速生成目標檢測數據集，工作效率高。

技術領域

本發明涉及人工智能技術領域，尤其涉及一種目標檢測數據集生成方法、裝置、系統和電子設備。

背景技術

在人工智能領域，基于深度學習的目標檢測算法是近年來比較熱門的研究課題。在把目標檢測算法應用到實際工業時，因為產品迭代周期短和數據集的短缺，需要為經常更新的目標檢測物體建立相對應的數據集。一般的建立目標檢測的數據集的過程主要有三步：

第一步，將待檢測物體放在不同的背景環境下，從盡可能多的不同角度拍照；

第二步，人工地用標簽軟件將待檢測目標物從放置臺背景用四方形框出并標簽；

第三步，將被編輯的圖片保存為二進制文件，形成數據集，方便以后檢測算法讀入。

通常情況下，為了保證訓練出來模型的魯棒性，同一個被檢測的物體要有數千張到數萬張不等的數據集。用人工的方法來逐張地采集照片并對數據分割和標簽，會耗費大量的時間和精力，且效率太低不能滿足工業產品特別是電子產品更新換代的要求。

發明內容

鑒于上述現有技術的不足之處，本申請實施例的目的在于提供一種目標檢測數據集生成方法、裝置、系統和電子設備，其生成目標檢測數據集的效率高。

第一方面，本申請實施例提供一種目標檢測數據集生成方法，應用于數據采集裝置,所述數據采集裝置包括采集臺，所述采集臺包括可旋轉的轉臺和設置在所述轉臺上方且角度可調的攝像機，所述目標檢測數據集生成方法包括步驟：

獲取所述采集臺采集的多張不同角度的工件的第一圖片；

提取每張所述第一圖片中的工件圖像；

把所述工件圖像放入預存的背景圖片中，得到第二圖片；

根據工件圖像的位置和大小在所述第二圖片中生成邊框信息和標簽。

所述的目標檢測數據集生成方法中，所述通過所述采集臺獲取多張不同角度的工件的第一圖片的步驟包括：

向所述采集臺發送控制指令，以控制所述采集臺的轉臺轉動多個不同的角度，并在轉臺的每個角度下，控制所述攝像機從多個不同角度拍攝工件的第一圖片。

所述的目標檢測數據集生成方法中，所述提取每張所述第一圖片中的工件圖像的步驟之后，還包括：

對每個所述工件圖像進行預處理，從而由每個所述工件圖像生成多個角度和/或大小不同的工件圖像。

進一步的，所述對每個所述工件圖像進行預處理，從而由每個所述工件圖像生成多個角度和/或大小不同的工件圖像的步驟包括：

對所述工件圖像進行旋轉處理：把所述工件圖像轉動多個不同角度，每轉動一個角度則保存為一個新的工件圖像；

和/或

對所述工件圖像進行縮放處理：把所述工件圖像按多個縮放比例值進行縮放，每縮放一次則保存為一個新的工件圖像。

所述的目標檢測數據集生成方法中，所述把所述工件圖像放入預存的背景圖片中，得到第二圖片的步驟包括：

把同一個工件圖像在不同位置放入所述背景圖片，從而每個工件圖像對應得到多個第二圖片。

第二方面，本申請實施例提供一種目標檢測數據集生成裝置，包括：