[發(fā)明專(zhuān)利]一種目標(biāo)樣本標(biāo)簽的分配方法

在審

• 申請(qǐng)?zhí)枺?/td>: 202110965314.5
• 申請(qǐng)日: 2021-08-23
• 公開(kāi)（公告）號(hào): CN113657519A
• 公開(kāi)（公告）日: 2021-11-16
• 發(fā)明（設(shè)計(jì)）人: 朱曉東;劉國(guó)清;季思文
• 申請(qǐng)（專(zhuān)利權(quán)）人: 南京佑駕科技有限公司
• 主分類(lèi)號(hào): G06K9/62
• 分類(lèi)號(hào): G06K9/62;G06K9/46
• 代理公司: 南京縱橫知識(shí)產(chǎn)權(quán)代理有限公司 32224
• 代理人: 董建林

1.一種目標(biāo)樣本標(biāo)簽的分配方法，其特征在于，包括：

根據(jù)圖像中的目標(biāo)中心點(diǎn)確定初始標(biāo)簽區(qū)域；

提取圖像的特征圖信息，然后在特征圖的每一特征點(diǎn)上生成預(yù)選錨點(diǎn)，初步篩選出初始標(biāo)簽區(qū)域內(nèi)的錨點(diǎn)；

使用分類(lèi)損失函數(shù)和位置回歸損失函數(shù)計(jì)算初始標(biāo)簽區(qū)域內(nèi)的每個(gè)錨點(diǎn)的分類(lèi)損失值和位置回歸損失值，將分類(lèi)損失值和位置回歸損失值按照相應(yīng)權(quán)重累加，得到各錨點(diǎn)的總損失值；

根據(jù)各錨點(diǎn)的總損失值選取最終錨點(diǎn)；

對(duì)選取的最終錨點(diǎn)進(jìn)行標(biāo)簽分配。

2.根據(jù)權(quán)利要求1所述的一種目標(biāo)樣本標(biāo)簽的分配方法，其特征在于，所述圖像的兩個(gè)對(duì)角點(diǎn)分別為(tl_x，tl_y)和(br_x，br_y)；所述初始標(biāo)簽區(qū)域的兩個(gè)對(duì)角點(diǎn)分別為(ctl_x，ctl_y)和(cbr_x，cbr_y)；所述圖像的兩個(gè)對(duì)角點(diǎn)與所述初始標(biāo)簽區(qū)域的兩個(gè)對(duì)角點(diǎn)的關(guān)系為：

其中n為調(diào)整初始標(biāo)簽區(qū)域大小的參數(shù)。

3.根據(jù)權(quán)利要求1所述的一種目標(biāo)樣本標(biāo)簽的分配方法，其特征在于，所述分類(lèi)損失函數(shù)為：

其中，a是錨點(diǎn)預(yù)測(cè)的概率值；錨點(diǎn)的標(biāo)記為目標(biāo)時(shí)，y為1，否則，y為0；x是所屬分類(lèi)下各錨點(diǎn),n'為所屬分類(lèi)下錨點(diǎn)的數(shù)量。

4.根據(jù)權(quán)利要求1所述的一種目標(biāo)樣本標(biāo)簽的分配方法，其特征在于，假設(shè)錨點(diǎn)的預(yù)測(cè)框?yàn)锳，錨點(diǎn)的標(biāo)注框?yàn)锽，位置回歸損失函數(shù)為：

其中IoU為預(yù)測(cè)框A與標(biāo)注框B的交并比,ρ²(b,b^gt)代表A與B中心點(diǎn)之間的歐幾里得距離，c為A與B之間最小外接矩形測(cè)對(duì)角線的長(zhǎng)度，v表示A與B之間長(zhǎng)寬比的距離，α是權(quán)重系數(shù)。

5.根據(jù)權(quán)利要求4所述的一種目標(biāo)樣本標(biāo)簽的分配方法，其特征在于，v的計(jì)算公式為：

w^gt為標(biāo)注框的寬度，h^gt為標(biāo)注框的高度，w為預(yù)測(cè)框的寬度，h為預(yù)測(cè)框的高度。

6.根據(jù)權(quán)利要求5所述的一種目標(biāo)樣本標(biāo)簽的分配方法，其特征在于，權(quán)重系數(shù)α為：

7.根據(jù)權(quán)利要求1所述的一種目標(biāo)樣本標(biāo)簽的分配方法，其特征在于，根據(jù)總損失值選取最終錨點(diǎn)，包括：根據(jù)總損失值對(duì)初始標(biāo)簽區(qū)域內(nèi)的各錨點(diǎn)進(jìn)行排序，由大到小選取設(shè)定數(shù)量a的錨點(diǎn)。

8.根據(jù)權(quán)利要求7所述的一種目標(biāo)樣本標(biāo)簽的分配方法，其特征在于，累加選取的設(shè)定數(shù)量a的錨點(diǎn)的總損失值并取整為數(shù)量b，作為重新選取錨點(diǎn)的數(shù)量，根據(jù)初始標(biāo)簽區(qū)域內(nèi)各錨點(diǎn)的排序，由大到小選取數(shù)量b的錨點(diǎn)。

9.根據(jù)權(quán)利要求1、7或8所述的一種目標(biāo)樣本標(biāo)簽的分配方法，其特征在于，響應(yīng)于有錨點(diǎn)重疊分配時(shí)，選擇保留其中分類(lèi)損失值最大的錨點(diǎn)，去除與分類(lèi)損失值最大的錨點(diǎn)重疊的錨點(diǎn)。