[發(fā)明專(zhuān)利]一種基于圖像樣本的自動(dòng)數(shù)據(jù)增強(qiáng)策略選擇方法

在審

• 申請(qǐng)?zhí)枺?/td>: 202011526140.4
• 申請(qǐng)日: 2020-12-22
• 公開(kāi)（公告）號(hào): CN112651892A
• 公開(kāi)（公告）日: 2021-04-13
• 發(fā)明（設(shè)計(jì)）人: 徐夢(mèng)娟;姚鵬;申書(shū)偉;邵鵬飛
• 申請(qǐng)（專(zhuān)利權(quán)）人: 中國(guó)科學(xué)技術(shù)大學(xué)
• 主分類(lèi)號(hào): G06T5/00
• 分類(lèi)號(hào): G06T5/00;G06T7/00;G06T7/90;G06T7/11;G06N3/04;G06N3/08
• 代理公司: 北京科迪生專(zhuān)利代理有限責(zé)任公司 11251
• 代理人: 顧煒

1.一種基于圖像樣本的自動(dòng)數(shù)據(jù)增強(qiáng)策略選擇方法，其特征在于，包括如下步驟：

步驟1、獲取目標(biāo)圖像數(shù)據(jù)集；

步驟2、建立數(shù)據(jù)增強(qiáng)策略搜索空間，所述搜索空間包括多種數(shù)據(jù)增強(qiáng)策略，同時(shí)設(shè)立每種增強(qiáng)策略的量級(jí)取值范圍；

步驟3、設(shè)定參數(shù)N值，其中N值表示對(duì)每一個(gè)圖像隨機(jī)進(jìn)行N次圖像數(shù)據(jù)轉(zhuǎn)換操作，每次操作都均等的從所設(shè)立的搜索空間中選擇一種增強(qiáng)策略進(jìn)行圖像變換操作；

步驟4、設(shè)定參數(shù)P值，其中P值表示每一個(gè)增強(qiáng)策略執(zhí)行的概率值，即單次增強(qiáng)策略有P的概率值會(huì)進(jìn)行增強(qiáng)操作，也有1-P的概率保持圖像不變，若進(jìn)行圖像變換，則變換幅度在設(shè)定變換量級(jí)范圍內(nèi)隨機(jī)選擇；

步驟5、在經(jīng)過(guò)N次圖像變換后，以隨機(jī)裁剪的方式對(duì)經(jīng)過(guò)圖像增強(qiáng)的數(shù)據(jù)進(jìn)行剪裁，獲取輸入至神經(jīng)網(wǎng)絡(luò)模型的圖像樣本；

步驟6、將裁剪后的圖像數(shù)據(jù)輸入到深度神經(jīng)網(wǎng)絡(luò)模型里面進(jìn)行進(jìn)一步訓(xùn)練，以便更好的提取圖像特征并進(jìn)一步應(yīng)用。

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

3.根據(jù)權(quán)利要求1所述的一種基于圖像樣本的自動(dòng)數(shù)據(jù)增強(qiáng)策略選擇方法，其特征在于，所述步驟3中，當(dāng)N1時(shí)，對(duì)一個(gè)訓(xùn)練樣本，均勻的進(jìn)行顏色和形狀變換，即首先從顏色變換大類(lèi)中隨機(jī)選擇增強(qiáng)策略進(jìn)行變換，然后從形狀變換中隨機(jī)選擇增強(qiáng)策略變換。

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