[發(fā)明專利]物體檢測(cè)模型訓(xùn)練方法及裝置

在審

• 申請(qǐng)?zhí)枺?/td>: 201910659672.6
• 申請(qǐng)日: 2019-07-19
• 公開（公告）號(hào): CN112241675A
• 公開（公告）日: 2021-01-19
• 發(fā)明（設(shè)計(jì)）人: 周定富;方進(jìn);宋希彬;官晨曄;楊睿剛
• 申請(qǐng)（專利權(quán)）人: 百度在線網(wǎng)絡(luò)技術(shù)（北京）有限公司
• 主分類號(hào): G06K9/00
• 分類號(hào): G06K9/00;G06K9/62;G06N3/08
• 代理公司: 北京市鑄成律師事務(wù)所 11313
• 代理人: 楊瑾瑾;林蕾

本發(fā)明實(shí)施例提出一種物體檢測(cè)模型訓(xùn)練方法及裝置。所述方法包括：根據(jù)檢測(cè)目標(biāo)的標(biāo)注數(shù)據(jù)確定物體真值包圍框；將所述檢測(cè)目標(biāo)輸入物體檢測(cè)模型，并根據(jù)所述物體檢測(cè)模型輸出的檢測(cè)數(shù)據(jù)得到物體檢測(cè)包圍框；根據(jù)所述物體真值包圍框和所述物體檢測(cè)包圍框，確定所述物體真值包圍框和所述物體檢測(cè)包圍框的交并比；根據(jù)所述交并比，確定損失函數(shù)值；根據(jù)所述損失函數(shù)值，反向傳播優(yōu)化所述物體檢測(cè)模型。本發(fā)明實(shí)施例提供的物體檢測(cè)模型訓(xùn)練方法及裝置，能夠提高物體檢測(cè)模型檢測(cè)的準(zhǔn)確率。

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

本發(fā)明涉及計(jì)算機(jī)技術(shù)領(lǐng)域，尤其涉及一種物體檢測(cè)模型的訓(xùn)練方法及裝置。

背景技術(shù)

物體檢測(cè)是計(jì)算機(jī)視覺領(lǐng)域的重要課題，在虛擬現(xiàn)實(shí)、文物保護(hù)、機(jī)械加工、計(jì)算機(jī)仿真等領(lǐng)域有著廣泛的應(yīng)用。無人駕駛是目前交通運(yùn)輸領(lǐng)域的一項(xiàng)新技術(shù)，具有廣闊的前景。在無人駕駛技術(shù)中，物體檢測(cè)技術(shù)起到了非常重要的作用。為了實(shí)現(xiàn)智能駕駛中的物體檢測(cè)，通常會(huì)采用深度學(xué)習(xí)模型，利用一些現(xiàn)有數(shù)據(jù)取訓(xùn)練深度學(xué)習(xí)模型，使得模型能夠在訓(xùn)練過程中學(xué)習(xí)如何識(shí)別物體。基于安全性的考慮，駕駛技術(shù)中的物體檢測(cè)技術(shù)需要具有極高的準(zhǔn)確性。

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

本發(fā)明實(shí)施例提供一種物體檢測(cè)模型訓(xùn)練方法，以解決現(xiàn)有技術(shù)中的一個(gè)或多個(gè)技術(shù)問題。

第一方面，本發(fā)明實(shí)施例提供了一種物體檢測(cè)模型訓(xùn)練方法，包括：

根據(jù)檢測(cè)目標(biāo)的標(biāo)注數(shù)據(jù)確定物體真值包圍框；

將所述檢測(cè)目標(biāo)輸入物體檢測(cè)模型，并根據(jù)所述物體檢測(cè)模型輸出的檢測(cè)數(shù)據(jù)得到物體檢測(cè)包圍框；

根據(jù)所述物體真值包圍框和所述物體檢測(cè)包圍框，確定所述物體真值包圍框和所述物體檢測(cè)包圍框的交并比；

根據(jù)所述交并比，確定損失函數(shù)值；

根據(jù)所述損失函數(shù)值，優(yōu)化所述物體檢測(cè)模型。

在一種實(shí)施方式中，在所述物體真值包圍框和所述物體檢測(cè)包圍框之間存在交疊的情況下，根據(jù)所述交并比，確定損失函數(shù)值，包括：

根據(jù)下述公式，確定損失函數(shù)值：

L＝1-IoU；

其中L為所述損失函數(shù)值，IoU為所述交并比。

在一種實(shí)施方式中，在所述物體真值包圍框和所述物體檢測(cè)包圍框之間不存在交疊的情況下，根據(jù)所述交并比，確定損失函數(shù)值，包括：

根據(jù)下述公式，確定損失函數(shù)值：

L＝1-GIoU；

其中，

其中，Area_C為包含所述物體真值包圍框和所述物體檢測(cè)包圍框的最小包圍框的面積，Area_U為所述物體真值包圍框和所述物體檢測(cè)包圍框的并集的面積，IoU為所述交并比。

在一種實(shí)施方式中，根據(jù)所述損失函數(shù)值，優(yōu)化所述物體檢測(cè)模型，包括：

根據(jù)所述損失函數(shù)值，進(jìn)行反向傳播計(jì)算，獲得反向傳播計(jì)算梯度值；

根據(jù)所述反向傳播計(jì)算梯度值，優(yōu)化所述物體檢測(cè)模型的參數(shù)。

在一種實(shí)施方式中，根據(jù)所述物體真值包圍框和所述物體檢測(cè)包圍框，確定所述物體真值包圍框和所述物體檢測(cè)包圍框的交并比，包括：

確定所述物體檢測(cè)包圍框和所述物體真值包圍框的交集和并集；

根據(jù)所述交集和所述并集，確定所述交并比。

在一種實(shí)施方式中，所述物體真值包圍框的任意邊與所述物體檢測(cè)包圍框的任意邊處于非平行狀態(tài)。