[發(fā)明專利]基于顏色的數(shù)據(jù)增強(qiáng)方法和系統(tǒng)及計(jì)算機(jī)設(shè)備和存儲(chǔ)介質(zhì)

在審

• 申請(qǐng)?zhí)枺?/td>: 201911318909.0
• 申請(qǐng)日: 2019-12-19
• 公開（公告）號(hào): CN111062862A
• 公開（公告）日: 2020-04-24
• 發(fā)明（設(shè)計(jì)）人: 黃俊清;艾莎亞·巴拉穆魯甘;申省梅;馬原
• 申請(qǐng)（專利權(quán)）人: 北京澎思科技有限公司
• 主分類號(hào): G06T3/00
• 分類號(hào): G06T3/00;G06T7/11;G06T7/187
• 代理公司: 北京植德律師事務(wù)所 11780
• 代理人: 冉晶;唐華東

本公開的實(shí)施例提供了一種基于顏色的數(shù)據(jù)增強(qiáng)方法和系統(tǒng)及裝置。方法包括：獲取輸入圖像；對(duì)所述輸入圖像進(jìn)行ROI分割，提取目標(biāo)物圖像；通過基于RGB規(guī)則的顏色轉(zhuǎn)換，將所述目標(biāo)物圖像從源色轉(zhuǎn)換為一種或多種不同目標(biāo)顏色的純色圖像；以及，將所述輸入圖像轉(zhuǎn)換為保持紋理的灰度圖像；將所述灰度圖像的紋理特征添加到所述純色圖像中，進(jìn)行紋理歸一化，生成一個(gè)或多個(gè)具有不同目標(biāo)顏色的輸出圖像。本公開實(shí)施例所提供的數(shù)據(jù)增強(qiáng)方法將圖像中目標(biāo)物的源色擴(kuò)增為多種目標(biāo)顏色，從而在保持原始圖像質(zhì)量的情況下對(duì)圖像樣本數(shù)量進(jìn)行擴(kuò)充，同時(shí)，降低生成新圖像的時(shí)間，降低收集新數(shù)據(jù)的成本，以此提高深度學(xué)習(xí)模型的性能。

技術(shù)領(lǐng)域

本公開涉及計(jì)算機(jī)視覺技術(shù)領(lǐng)域，具體涉及一種基于顏色的數(shù)據(jù)增強(qiáng)方法和系統(tǒng)及計(jì)算機(jī)設(shè)備和存儲(chǔ)介質(zhì)。

背景技術(shù)

當(dāng)前，基于神經(jīng)網(wǎng)絡(luò)的深度學(xué)習(xí)技術(shù)已經(jīng)在諸如物體分類、圖像識(shí)別、文本處理、、面部識(shí)別、自動(dòng)駕駛、人機(jī)對(duì)話等領(lǐng)域取得了巨大進(jìn)展。隨著神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu)的加深和算法的提升，對(duì)用于訓(xùn)練神經(jīng)網(wǎng)絡(luò)的數(shù)據(jù)樣本的需求量越來(lái)越大。尤其是在與圖像相關(guān)的神經(jīng)網(wǎng)絡(luò)中，較差的圖像或較少的圖像數(shù)量，可能導(dǎo)致所訓(xùn)練出的神經(jīng)網(wǎng)絡(luò)模型僅在訓(xùn)練域上理想的泛化能力，而在實(shí)際使用中不理想，進(jìn)而導(dǎo)致最終產(chǎn)品(例如監(jiān)視系統(tǒng))的工作性能受到不利影響。

發(fā)明內(nèi)容

第一方面，本公開的實(shí)施例提供一種基于顏色的數(shù)據(jù)增強(qiáng)方法，包括：獲取輸入圖像；對(duì)所述輸入圖像進(jìn)行ROI分割，提取目標(biāo)物圖像；通過基于RGB規(guī)則的顏色轉(zhuǎn)換，將所述目標(biāo)物圖像從源色轉(zhuǎn)換為一種或多種不同目標(biāo)顏色的純色圖像；以及，將所述輸入圖像轉(zhuǎn)換為保持紋理的灰度圖像；將所述灰度圖像的紋理特征添加到所述純色圖像中，進(jìn)行紋理歸一化，生成一個(gè)或多個(gè)具有不同目標(biāo)顏色的輸出圖像。

在一些實(shí)施例中，所述的對(duì)所述輸入圖像進(jìn)行ROI分割，包括：基于輸入圖像的形狀和顏色值，使用語(yǔ)義分割方法進(jìn)行ROI分割。

在一些實(shí)施例中，所述的基于RGB規(guī)則的顏色轉(zhuǎn)換，包括：在RGB規(guī)則的搜索空間中，通過調(diào)整各個(gè)色彩通道的值實(shí)現(xiàn)顏色轉(zhuǎn)換。

在一些實(shí)施例中，所述的將所述灰度圖像的紋理特征添加到所述純色圖像中，進(jìn)行紋理歸一化，包括：將純色圖像與灰度圖像進(jìn)行疊加平均處理。

第二方面，本公開的實(shí)施例提供一種基于顏色的數(shù)據(jù)增強(qiáng)系統(tǒng)，包括：圖像輸入模塊，用于獲取輸入圖像；ROI分割模塊，用于對(duì)所述輸入圖像進(jìn)行ROI分割，提取目標(biāo)物圖像；顏色轉(zhuǎn)換模塊，用于通過基于RGB規(guī)則的顏色轉(zhuǎn)換，將所述目標(biāo)物圖像從源色轉(zhuǎn)換為一種或多種不同目標(biāo)顏色的純色圖像；灰度轉(zhuǎn)換模塊，用于將所述輸入圖像轉(zhuǎn)換為保持紋理的灰度圖像；紋理歸一化模塊，用于將所述灰度圖像的紋理特征添加到所述純色圖像中，進(jìn)行紋理歸一化，生成一個(gè)或多個(gè)具有不同目標(biāo)顏色的輸出圖像；圖像輸出模塊，用于輸出所述輸出圖像。

在一些實(shí)施例中，所述ROI分割模塊，具體用于基于輸入圖像的形狀和顏色值，使用語(yǔ)義分割方法進(jìn)行ROI分割。

在一些實(shí)施例中，所述顏色轉(zhuǎn)換模塊，具體用于在RGB規(guī)則的搜索空間中，通過調(diào)整各個(gè)色彩通道的值實(shí)現(xiàn)顏色轉(zhuǎn)換。

在一些實(shí)施例中，所述紋理歸一化模塊，具體用于將純色圖像與灰度圖像進(jìn)行疊加平均處理。

第三方面，本公開的實(shí)施例提供一種計(jì)算機(jī)設(shè)備，包括

存儲(chǔ)器，用于非暫時(shí)性存儲(chǔ)計(jì)算機(jī)可讀指令；以及

處理器，用于運(yùn)行所述計(jì)算機(jī)可讀指令，所述計(jì)算機(jī)可讀指令被所述處理器運(yùn)行時(shí)執(zhí)行上述任一項(xiàng)實(shí)施例所述的方法。

第四方面，本公開的實(shí)施例提供一種存儲(chǔ)介質(zhì)，非暫時(shí)性地存儲(chǔ)計(jì)算機(jī)可讀指令，當(dāng)所述計(jì)算機(jī)可讀指令由計(jì)算機(jī)執(zhí)行時(shí)可以執(zhí)行上述任一項(xiàng)實(shí)施例所述的方法。

在本公開的一些可行的實(shí)施方式中，至少部分地取得了以下有益效果：