[發明專利]一種包含多目標的圖片數據集的生成方法及系統

在審

• 申請號: 202010556085.7
• 申請日: 2020-06-17
• 公開（公告）號: CN111882508A
• 公開（公告）日: 2020-11-03
• 發明（設計）人: 程德心;周風明;郝江波;周昭暉
• 申請（專利權）人: 武漢光庭信息技術股份有限公司
• 主分類號: G06T5/30
• 分類號: G06T5/30;G06T7/13;G06T7/12;G06T5/00;G06F16/951
• 代理公司: 武漢藍寶石專利代理事務所(特殊普通合伙) 42242
• 代理人: 嚴超

1.一種包含多目標的圖片數據集的生成方法，其特征在于，所述方法包括：

步驟1，基于網絡爬蟲爬取獲得包含目標對象的各個目標圖片；

步驟2，查找所述目標圖片中的目標對象的輪廓，通過像素運算將所述目標對象從所述目標圖片進行區分；

步驟3，根據需求圖片的目標對象的數量進行圖片合成。

2.根據權利要求1所述的方法，其特征在于，所述步驟1包括：

步驟101，采用廣度優先的爬蟲策略去爬取包含目標對象的圖片；

步驟102，采用密度聚類算法對爬取的圖片進行無監督簇聚類；

步驟103，采用特征識別的方法判斷圖片是否包含目標對象。

3.根據權利要求1所述的方法，其特征在于，所述步驟1之后還包括使用計算機視覺庫對圖片進行預處理，包括：高斯模糊、圖片灰度化、圖片的邊緣劃分、二值化處理、膨脹腐蝕操作以及中值濾波。

4.根據權利要求3所述的方法，其特征在于，所述利用sobel邊緣檢測方法進行圖片的邊緣劃分，根據鄰點加權差，在邊緣處達到極值時進行檢測。

5.根據權利要求1所述的方法，其特征在于，所述步驟2中采用輪廓檢測查找所述目標圖片中的目標對象的輪廓輪廓，通過PIL將所述目標對象從所述目標圖片進行區分，使所述目標圖片中除了所述目標對象之外的區域的像素點的RGB值均為0。

6.根據權利要求1所述的方法，其特征在于，所述步驟2還包括所述目標圖片不需要保留多角度圖片數據時對所述目標圖片進行透視變換，將其他角度的所述目標對象進行平面化處理。

7.根據權利要求1所述的方法，其特征在于，所述步驟3中，使用PIL進行圖片合成后，對所述合成后的圖片進行標簽信息整合。

8.一種包含多目標的圖片數據集的生成系統，其特征在于，所述系統包括：目標圖片查找模塊、目標圖片處理模塊和圖片合成模塊；

所述目標圖片查找模塊，用于基于網絡爬蟲爬取獲得包含目標對象的各個目標圖片；

所述目標圖片處理模塊，用于查找所述目標圖片中的目標對象的輪廓，通過像素運算將所述目標對象從所述目標圖片進行區分；

所述圖片合成模塊，用于根據需求圖片的目標對象的數量進行圖片合成。

9.一種電子設備，包括存儲器、處理器及存儲在存儲器上并可在處理器上運行的計算機程序，其特征在于，所述處理器執行所述程序時實現如權利要求1至7任一項所述包含多目標的圖片數據集的生成方法的步驟。

10.一種非暫態計算機可讀存儲介質，其上存儲有計算機程序，其特征在于，該計算機程序被處理器執行時實現如權利要求1至7任一項所述包含多目標的圖片數據集的生成方法的步驟。