[發(fā)明專利]特征提取方法、裝置及電子設(shè)備

在審

• 申請(qǐng)?zhí)枺?/td>: 202011176076.1
• 申請(qǐng)日: 2020-10-28
• 公開（公告）號(hào): CN112418243A
• 公開（公告）日: 2021-02-26
• 發(fā)明（設(shè)計(jì)）人: 莊月清;李伯勛;張弛
• 申請(qǐng)（專利權(quán)）人: 北京邁格威科技有限公司
• 主分類號(hào): G06K9/46
• 分類號(hào): G06K9/46;G06N3/04
• 代理公司: 北京超凡宏宇專利代理事務(wù)所(特殊普通合伙) 11463
• 代理人: 何少巖

本發(fā)明提供了一種特征提取方法、裝置及電子設(shè)備，涉及圖像處理的技術(shù)領(lǐng)域，包括：獲取待提取特征的目標(biāo)圖像；獲取目標(biāo)圖像對(duì)應(yīng)的掩膜圖像；其中，掩膜圖像與目標(biāo)圖像中的感興趣區(qū)域?qū)?yīng)；將目標(biāo)圖像和掩膜圖像輸入至特征提取網(wǎng)絡(luò)，通過特征提取網(wǎng)絡(luò)基于掩膜圖像對(duì)目標(biāo)圖像進(jìn)行特征提取。本發(fā)明可以有效提高特征提取效率。

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

本發(fā)明涉及圖像處理技術(shù)領(lǐng)域，尤其是涉及一種特征提取方法、裝置及電子設(shè)備。

背景技術(shù)

在現(xiàn)有的圖像識(shí)別等任務(wù)中大多涉及到特征提取操作，以提取出圖像的特征信息。目前，在對(duì)圖像執(zhí)行特征提取的操作前，大多需要通過檢測(cè)的方式將圖像中的物體所在區(qū)域摳取出來，然后對(duì)摳取得到的物體所在區(qū)域進(jìn)行特征提取，基于此，特征提取效率往往依賴于摳圖效率，摳圖效率已成為特征提取效率的制約瓶頸。

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

有鑒于此，本發(fā)明的目的在于提供一種特征提取方法、裝置及電子設(shè)備，無需摳圖即可執(zhí)行特征提取操作，可以有效提高特征提取效率。

為了實(shí)現(xiàn)上述目的，本發(fā)明實(shí)施例采用的技術(shù)方案如下：

第一方面，本發(fā)明實(shí)施例提供了一種特征提取方法，包括：獲取待提取特征的目標(biāo)圖像；獲取所述目標(biāo)圖像對(duì)應(yīng)的掩膜圖像；其中，所述掩膜圖像與所述目標(biāo)圖像中感興趣區(qū)域?qū)?yīng)；將所述目標(biāo)圖像和所述掩膜圖像輸入至特征提取網(wǎng)絡(luò)，通過所述特征提取網(wǎng)絡(luò)基于所述掩膜圖像對(duì)所述目標(biāo)圖像進(jìn)行特征提取。

進(jìn)一步，所述特征提取網(wǎng)絡(luò)包括多個(gè)依次連接的卷積層；所述通過所述特征提取網(wǎng)絡(luò)基于所述掩膜圖像對(duì)所述目標(biāo)圖像進(jìn)行特征提取的步驟，包括：對(duì)于所述特征提取網(wǎng)絡(luò)中的每個(gè)所述卷積層，確定該卷積層對(duì)應(yīng)的輸入特征圖，并通過該卷積層對(duì)其輸入特征圖進(jìn)行特征提取，得到該卷積層對(duì)應(yīng)的輸出特征圖；其中，所述特征提取網(wǎng)絡(luò)中第一卷積層對(duì)應(yīng)的輸入特征圖與所述目標(biāo)圖像和所述掩膜圖像相關(guān)，所述特征提取網(wǎng)絡(luò)中第二卷積層對(duì)應(yīng)的輸入特征圖與前一卷積層的輸出特征圖和所述掩膜圖像相關(guān)；根據(jù)所述特征提取網(wǎng)絡(luò)的輸出特征圖確定所述目標(biāo)圖像中感興趣區(qū)域的特征。

進(jìn)一步，所述確定該卷積層對(duì)應(yīng)的輸入特征圖的步驟，包括：當(dāng)該卷積層為所述第一卷積層時(shí)，將所述目標(biāo)圖像與所述掩膜圖像進(jìn)行按位相乘，得到該第一卷積層對(duì)應(yīng)的輸入特征圖；當(dāng)該卷積層為所述第二卷積層，將該第二卷積層的前一卷積層的輸出特征圖與所述掩膜圖像進(jìn)行按位相乘，得到該第二卷積層對(duì)應(yīng)的輸入特征圖。

進(jìn)一步，所述感興趣區(qū)域中包括至少一個(gè)目標(biāo)對(duì)象；所述根據(jù)所述特征提取網(wǎng)絡(luò)的輸出特征圖確定所述目標(biāo)圖像中感興趣區(qū)域的特征的步驟，包括：確定所述目標(biāo)圖像中各個(gè)所述目標(biāo)對(duì)象的目標(biāo)特征點(diǎn)；獲取所述特征提取網(wǎng)絡(luò)的輸出特征圖中各個(gè)所述目標(biāo)對(duì)象的目標(biāo)特征點(diǎn)對(duì)應(yīng)的特征；將各個(gè)所述目標(biāo)對(duì)象的目標(biāo)特征點(diǎn)對(duì)應(yīng)的特征均確定為所述目標(biāo)圖像中感興趣區(qū)域的特征。

進(jìn)一步，所述獲取所述特征提取網(wǎng)絡(luò)的輸出特征圖中各個(gè)所述目標(biāo)對(duì)象的目標(biāo)特征點(diǎn)對(duì)應(yīng)的特征的步驟，包括：對(duì)于每個(gè)所述目標(biāo)對(duì)象的目標(biāo)特征點(diǎn)，將所述特征提取網(wǎng)絡(luò)的輸出特征圖與該目標(biāo)對(duì)象的目標(biāo)特征點(diǎn)進(jìn)行按位相乘，得到該目標(biāo)對(duì)象的目標(biāo)特征點(diǎn)對(duì)應(yīng)的特征。

進(jìn)一步，所述確定所述目標(biāo)圖像中各個(gè)所述目標(biāo)對(duì)象的目標(biāo)特征點(diǎn)的步驟，包括：對(duì)所述目標(biāo)圖像進(jìn)行對(duì)象檢測(cè)，得到所述目標(biāo)圖像中的各個(gè)所述目標(biāo)對(duì)象；對(duì)于每個(gè)所述目標(biāo)對(duì)象，確定該目標(biāo)對(duì)象的目標(biāo)特征點(diǎn)。

進(jìn)一步，所述獲取所述目標(biāo)圖像對(duì)應(yīng)的掩膜圖像的步驟，包括：根據(jù)所述目標(biāo)圖像的尺寸建立初始圖像；其中，所述初始圖像的像素為第一數(shù)值；確定所述目標(biāo)圖像中的感興趣區(qū)域；將所述初始圖像中與所述目標(biāo)圖像的感興趣區(qū)域?qū)?yīng)的區(qū)域內(nèi)的像素從所述第一數(shù)值調(diào)整為第二數(shù)值，得到所述目標(biāo)圖像對(duì)應(yīng)的掩膜圖像。