[發明專利]一種基于圖像樣本的自動數據增強策略選擇方法

在審

• 申請號: 202011526140.4
• 申請日: 2020-12-22
• 公開（公告）號: CN112651892A
• 公開（公告）日: 2021-04-13
• 發明（設計）人: 徐夢娟;姚鵬;申書偉;邵鵬飛
• 申請（專利權）人: 中國科學技術大學
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06T7/00;G06T7/90;G06T7/11;G06N3/04;G06N3/08
• 代理公司: 北京科迪生專利代理有限責任公司 11251
• 代理人: 顧煒

1.一種基于圖像樣本的自動數據增強策略選擇方法，其特征在于，包括如下步驟：

步驟1、獲取目標圖像數據集；

步驟2、建立數據增強策略搜索空間，所述搜索空間包括多種數據增強策略，同時設立每種增強策略的量級取值范圍；

步驟3、設定參數N值，其中N值表示對每一個圖像隨機進行N次圖像數據轉換操作，每次操作都均等的從所設立的搜索空間中選擇一種增強策略進行圖像變換操作；

步驟4、設定參數P值，其中P值表示每一個增強策略執行的概率值，即單次增強策略有P的概率值會進行增強操作，也有1-P的概率保持圖像不變，若進行圖像變換，則變換幅度在設定變換量級范圍內隨機選擇；

步驟5、在經過N次圖像變換后，以隨機裁剪的方式對經過圖像增強的數據進行剪裁，獲取輸入至神經網絡模型的圖像樣本；

步驟6、將裁剪后的圖像數據輸入到深度神經網絡模型里面進行進一步訓練，以便更好的提取圖像特征并進一步應用。

2.根據權利要求1所述的一種基于圖像樣本的自動數據增強策略選擇方法，其特征在于，所述步驟2中的搜索空間包括多種數據增強操作，同時設立每種增強策略的增強量級的取值范圍，使每次變換幅度都在允許范圍內隨機選擇，同時將所有增強操作分為兩類，一類是改變圖像數據中的像素值，稱之為顏色變換，另一類是使圖像數據的形狀發生改變，稱之為形狀變換。

3.根據權利要求1所述的一種基于圖像樣本的自動數據增強策略選擇方法，其特征在于，所述步驟3中，當N1時，對一個訓練樣本，均勻的進行顏色和形狀變換，即首先從顏色變換大類中隨機選擇增強策略進行變換，然后從形狀變換中隨機選擇增強策略變換。

4.根據權利要求1所述的一種基于圖像樣本的自動數據增強策略選擇方法，其特征在于，所述的圖像變換包括為顏色變換和形狀變換兩個子集，其中顏色變換包括飽和度、對比度、高斯噪聲、圖像融合、亮度、清晰度、通道偏差、色彩均化、在YUV空間的色彩均化、色調分離、自動對比度、曝光、漸暈的變換；所述形狀變換包括旋轉、翻轉、錯切、畸變、同比例縮放、不同方向的縮放、剪切、隨機裁剪的變換。