[發明專利]圖片擴增與批注方法、裝置及計算機存儲介質

在審

• 申請號: 201910141909.1
• 申請日: 2019-02-26
• 公開（公告）號: CN111611421A
• 公開（公告）日: 2020-09-01
• 發明（設計）人: 胡博章
• 申請（專利權）人: 鴻富錦精密工業（武漢）有限公司;鴻海精密工業股份有限公司
• 主分類號: G06F16/535
• 分類號: G06F16/535;G06F16/50
• 代理公司: 深圳市賽恩倍吉知識產權代理有限公司 44334
• 代理人: 劉永輝;李艷霞

本發明提出一種圖片擴增與批注方法，具體包括如下步驟：建立原始圖片集；其中，原始圖片集內存儲有待擴增的原始圖片；翻轉旋轉所述原始圖片，得到擴增圖片并組成擴增圖片集；批注原始圖片；按照預設的轉換規則對批注后的原始圖片進行批注轉換得到批注后的擴增圖片；合并批注后的原始圖片和擴增圖片至訓練圖片集。本發明還對應提供了一種圖片擴增與批注裝置和計算機存儲介質，利用本發明的技術方案，先擴增圖片數量再統一圖片批注格式，將統一批注后的原始圖片和擴增圖片用于訓練AI深度學習模型，可以簡單有效地提高AI深度學習模型的準確度。

技術領域

本發明涉及數據處理技術領域，特別是一種圖片擴增與批注方法、裝置及計算機存儲介質。

背景技術

隨著現代社會的發展，使得計算機技術以及相應的軟件產品一直處于增長的狀態。各行各業在生產過程中均需要使用軟件產品，面對龐大的軟件產品需求。例如，從產線取得的AI模型訓練的圖片受限于設備、成本、時間壓力等因素使得用于訓練的圖片數量可能會不夠多，進而降低AI深度學習模型的準確度，用于AI深度學習模型的訓練圖片需統一批注格式。

發明內容

鑒于上述狀況，有必要提供一種圖片擴增與批注方法、裝置及計算機存儲介質、裝置及計算機存儲介質，通過擴增圖片數量和統一圖片批注格式以提高AI深度學習模型的準確度，以解決上述問題。

本發明第一方面提供了一種圖片擴增與批注方法，用于擴增圖片數量與批注圖片，所述圖片擴增與批注包括如下步驟：建立原始圖片集；其中，所述原始圖片集內存儲有待擴增的原始圖片；旋轉所述原始圖片，得到擴增圖片并組成擴增圖片集；批注所述原始圖片；按照預設的轉換規則對批注后的所述原始圖片進行批注轉換得到批注后的所述擴增圖片；合并批注后的所述原始圖片和所述擴增圖片至訓練圖片集。

優選地，步驟所述旋轉所述原始圖片，包括如下步驟：翻轉所述原始圖片得到翻轉圖片，使所述原始圖片與所述翻轉圖片形成鏡像對稱；以預設角度分別旋轉所述原始圖片和所述翻轉圖片對應得到第一擴增圖片和第二擴增圖片；其中，所述預設角度大于0°且小于360°。

優選地，所述原始圖片和所述翻轉圖片分別旋轉K次對應得到所述第一擴增圖片和所述第二擴增圖片，每次旋轉角度θ＝(2×360)/N度；其中，K＝(N/2)-1，N為圖片擴增數量的倍數，N為偶數。

優選地，所述批注所述原始圖片步驟包括批注部位的尺寸坐標、點坐標和格式坐標；其中，所述尺寸坐標為所述批注部位的面積在所述原始圖片面積中的水平和垂直方向的占比，所述點坐標為所述批注部位中其中一個點的點坐標，所述格式坐標為所述批注部位中水平方向坐標最小值、垂直方向坐標最小值、批注部位寬度和批注部位高度。

優選地，所述按照預設的轉換規則對批注后的所述原始圖片進行批注轉換得到批注后的所述擴增圖片步驟，包括：通過所述原始圖片的所述批注部位的尺寸坐標的坐標值的轉換得到所述擴增圖片的批注部位的尺寸坐標；通過所述原始圖片的所述批注部位的點坐標值與尺寸坐標值的轉換得到所述擴增圖片的批注部位的點坐標；通過所述原始圖片的所述批注部位的點坐標值與格式坐標值的轉換得到所述擴增圖片的批注部位的格式坐標。

本發明第二方面提供了一種圖片擴增與批注裝置，用于擴增圖片數量與批注圖片，所述圖片擴增與批注裝置包括：處理器；以及存儲器，所述存儲器中存儲有多個程序模塊，多個所述程序模塊由所述處理器運行并執行如下步驟：建立原始圖片集；其中，所述原始圖片集內存儲有待擴增的原始圖片；旋轉所述原始圖片，得到擴增圖片并組成擴增圖片集；批注所述原始圖片；按照預設的轉換規則對批注后的所述原始圖片進行批注轉換得到批注后的所述擴增圖片；合并批注后的所述原始圖片和所述擴增圖片至訓練圖片集。

優選地，多個所述程序模塊由所述處理器運行并還執行如下步驟：翻轉所述原始圖片得到翻轉圖片，使所述原始圖片與所述翻轉圖片形成鏡像對稱；以預設角度分別旋轉所述原始圖片和所述翻轉圖片對應得到第一擴增圖片和第二擴增圖片；其中，所述預設角度大于0°且小于360°。