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

在審

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

本發明涉及一種包含多目標的圖片數據集的生成方法及系統，基于網絡爬蟲爬取獲得包含目標對象的各個目標圖片；查找目標圖片中的目標對象的輪廓，通過像素運算將目標對象從目標圖片進行區分；根據需求圖片的目標對象的數量進行圖片合成結合計算機視覺技術，設定好識別目標圖片的算法條件下，利用圖片處理方法進行相應的圖片預處理，提取和合成，生成對應的多目標圖片數據集，相比于人工采集識別，提升了效率，降低了行業成本，針對不同環境下的圖片都能靈活處理。

技術領域

本發明涉及計算機視覺領域，尤其涉及一種包含多目標的圖片數據集的生成方法及系統。

背景技術

目前在計算機視覺領域中，已有很多開放的數據集，例如人臉數據集、車牌數據集合交通標志牌數據集，但是只能根據現有數據集進行選擇和使用，無法做到根據各種時刻方便的獲取數據集。

比如想要獲取單張圖片包含多目標的數據集，可能需要利用雙目攝像頭單獨去特定場景獲取圖片數據集。需要大量人工操作達到目的，同時也增加了行業成本。

發明內容

本發明針對現有技術中存在的技術問題，提供一種包含多目標的圖片數據集的生成方法，解決現有技術中無法方便獲取多目標圖片的數據集的問題。

本發明解決上述技術問題的技術方案如下：一種包含多目標的圖片數據集的生成方法，包括：

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

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

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

一種包含多目標的圖片數據集的生成系統，包括：目標圖片查找模塊、目標圖片處理模塊和圖片合成模塊；

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

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

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

本發明的有益效果是：本發明提供的一種包含多目標的圖片數據集的生成方法及系統，結合計算機視覺技術，設定好識別目標圖片的算法條件下，利用圖片處理方法進行相應的圖片預處理，提取和合成，生成對應的多目標圖片數據集，相比于人工采集識別，提升了效率，降低了行業成本，針對不同環境下的圖片都能靈活處理。

在上述技術方案的基礎上，本發明還可以做如下改進。

進一步，所述步驟1包括：

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

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

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

進一步，所述步驟1之后還包括使用計算機視覺庫對圖片進行預處理，包括：高斯模糊、圖片灰度化、圖片的邊緣劃分、二值化處理、膨脹腐蝕操作以及中值濾波。

進一步，所述利用sobel邊緣檢測方法進行圖片的邊緣劃分，根據鄰點加權差，在邊緣處達到極值時進行檢測。

進一步，所述步驟2中采用輪廓檢測查找所述目標圖片中的目標對象的輪廓輪廓，通過PIL將所述目標對象從所述目標圖片進行區分，使所述目標圖片中除了所述目標對象之外的區域的像素點的RGB值均為0。

進一步，所述步驟2還包括所述目標圖片不需要保留多角度圖片數據時對所述目標圖片進行透視變換，將其他角度的所述目標對象進行平面化處理。