[發(fā)明專利]目標(biāo)框生成方法、裝置、非易失性存儲(chǔ)介質(zhì)及計(jì)算機(jī)設(shè)備

在審

• 申請(qǐng)?zhí)枺?/td>: 202210536745.4
• 申請(qǐng)日: 2022-05-17
• 公開(kāi)（公告）號(hào): CN114926631A
• 公開(kāi)（公告）日: 2022-08-19
• 發(fā)明（設(shè)計(jì)）人: 朱亞旋;陳博;尹榮彬;薛鴻;徐名源
• 申請(qǐng)（專利權(quán)）人: 中國(guó)第一汽車股份有限公司
• 主分類號(hào): G06V10/25
• 分類號(hào): G06V10/25;G06V10/77;G06V10/82;G06N3/04;G06N3/08
• 代理公司: 北京博浩百睿知識(shí)產(chǎn)權(quán)代理有限責(zé)任公司 11134
• 代理人: 謝湘寧

1.一種目標(biāo)框生成方法，其特征在于，包括：

獲取目標(biāo)圖像的特征圖，其中，所述特征圖包括多個(gè)特征區(qū)域，每個(gè)所述特征區(qū)域?qū)?yīng)于所述目標(biāo)圖像中的一塊像素區(qū)域；

確定所述多個(gè)特征區(qū)域分別對(duì)應(yīng)的多個(gè)目標(biāo)置信度，其中，每個(gè)所述目標(biāo)置信度用于描述采用對(duì)應(yīng)的特征區(qū)域?qū)δ繕?biāo)框進(jìn)行預(yù)測(cè)的可靠程度，所述目標(biāo)框用于在所述目標(biāo)圖像中框出目標(biāo)對(duì)象；

根據(jù)所述多個(gè)目標(biāo)置信度，從所述多個(gè)特征區(qū)域中選取至少一個(gè)錨點(diǎn)區(qū)域，其中，所述至少一個(gè)錨點(diǎn)區(qū)域?yàn)樗龆鄠€(gè)特征區(qū)域中的特征區(qū)域；

基于所述至少一個(gè)錨點(diǎn)區(qū)域，預(yù)測(cè)所述目標(biāo)框在所述目標(biāo)圖像中的位置。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述確定所述多個(gè)特征區(qū)域分別對(duì)應(yīng)的多個(gè)目標(biāo)置信度，包括：

確定所述多個(gè)特征區(qū)域分別對(duì)應(yīng)的多個(gè)初始置信度和多個(gè)置信度閾值；

基于所述初始置信度和置信度閾值，確定所述多個(gè)特征區(qū)域分別對(duì)應(yīng)的所述多個(gè)目標(biāo)置信度。

3.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述確定所述多個(gè)特征區(qū)域包括的每個(gè)特征區(qū)域的初始置信度和置信度閾值，包括：

對(duì)所述特征圖進(jìn)行第一卷積運(yùn)算，得到所述多個(gè)特征區(qū)域分別對(duì)應(yīng)的多個(gè)第一特征值，其中，所述初始置信度包括所述第一特征值；

對(duì)所述特征圖進(jìn)行第二卷積運(yùn)算，得到所述多個(gè)特征區(qū)域分別對(duì)應(yīng)的多個(gè)第二特征值；

將所述多個(gè)第二特征值分別輸入目標(biāo)激活函數(shù)，得到所述多個(gè)置信度閾值。

4.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述目標(biāo)激活函數(shù)包括h-sigmoid函數(shù)，其中，所述h-sigmoid函數(shù)為基于relu6函數(shù)構(gòu)建的。

5.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述基于所述初始置信度和置信度閾值，確定所述多個(gè)特征區(qū)域分別對(duì)應(yīng)的所述多個(gè)目標(biāo)置信度，包括：

依次對(duì)所述多個(gè)特征區(qū)域中每一個(gè)特征區(qū)域?qū)?yīng)的初始置信度和置信度閾值進(jìn)行可微分二值化操作，得到所述多個(gè)特征區(qū)域分別對(duì)應(yīng)的所述多個(gè)目標(biāo)置信度。

6.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述基于所述至少一個(gè)錨點(diǎn)區(qū)域，預(yù)測(cè)所述目標(biāo)框在所述目標(biāo)圖像中的位置，包括：

根據(jù)所述特征圖，預(yù)測(cè)所述目標(biāo)對(duì)象分別在所述至少一個(gè)錨點(diǎn)區(qū)域中的類別占比；

根據(jù)所述類別占比和所述錨點(diǎn)區(qū)域，預(yù)測(cè)所述目標(biāo)框在所述目標(biāo)圖像中的位置以及所述目標(biāo)對(duì)象的類別。

7.根據(jù)權(quán)利要求6所述的方法，其特征在于，在目標(biāo)錨點(diǎn)區(qū)域?yàn)樗鲥^點(diǎn)區(qū)域中的任意一個(gè)區(qū)域的情況下，所述目標(biāo)對(duì)象在所述目標(biāo)錨點(diǎn)區(qū)域的類別占比，包括：所述目標(biāo)對(duì)象在所述目標(biāo)錨點(diǎn)區(qū)域?qū)?yīng)的像素區(qū)域中的像素占比。

8.一種目標(biāo)框生成裝置，其特征在于，包括：

獲取模塊，用于獲取目標(biāo)圖像的特征圖，其中，所述特征圖包括多個(gè)特征區(qū)域，每個(gè)所述特征區(qū)域?qū)?yīng)于所述目標(biāo)圖像中的一塊像素區(qū)域；

確定模塊，用于確定所述多個(gè)特征區(qū)域分別對(duì)應(yīng)的多個(gè)目標(biāo)置信度，其中，每個(gè)所述目標(biāo)置信度用于描述采用對(duì)應(yīng)的特征區(qū)域?qū)δ繕?biāo)框進(jìn)行預(yù)測(cè)的可靠程度，所述目標(biāo)框用于在所述目標(biāo)圖像中框出目標(biāo)對(duì)象；

選取模塊，用于根據(jù)所述多個(gè)目標(biāo)置信度，從所述多個(gè)特征區(qū)域中選取至少一個(gè)錨點(diǎn)區(qū)域，其中，所述至少一個(gè)錨點(diǎn)區(qū)域?yàn)樗龆鄠€(gè)特征區(qū)域中的特征區(qū)域；

預(yù)測(cè)模塊，用于基于所述至少一個(gè)錨點(diǎn)區(qū)域，預(yù)測(cè)所述目標(biāo)框在所述目標(biāo)圖像中的位置。

9.一種非易失性存儲(chǔ)介質(zhì)，其特征在于，所述非易失性存儲(chǔ)介質(zhì)包括存儲(chǔ)的程序，其中，在所述程序運(yùn)行時(shí)控制所述非易失性存儲(chǔ)介質(zhì)所在設(shè)備執(zhí)行權(quán)利要求1至7中任意一項(xiàng)所述目標(biāo)框生成方法。

10.一種計(jì)算機(jī)設(shè)備，其特征在于，所述計(jì)算機(jī)設(shè)備包括處理器，所述處理器用于運(yùn)行程序，其中，所述程序運(yùn)行時(shí)執(zhí)行權(quán)利要求1至7中任意一項(xiàng)所述目標(biāo)框生成方法。