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

在審

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

1.一種物體檢測模型訓練方法，其特征在于，包括：

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

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

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

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

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

2.根據(jù)權利要求1所述的方法，其特征在于，在所述物體真值包圍框和所述物體檢測包圍框之間存在交疊的情況下，根據(jù)所述交并比，確定損失函數(shù)值，包括：

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

L＝1-IoU；

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

3.根據(jù)權利要求1所述的方法，其特征在于，在所述物體真值包圍框和所述物體檢測包圍框之間不存在交疊的情況下，根據(jù)所述交并比，確定損失函數(shù)值，包括：

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

L＝1-GIoU；

其中，

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

4.根據(jù)權利要求1所述的方法，其特征在于，根據(jù)所述損失函數(shù)值，優(yōu)化所述物體檢測模型，包括：

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

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

5.根據(jù)權利要求1所述的方法，其特征在于，根據(jù)所述物體真值包圍框和所述物體檢測包圍框，確定所述物體真值包圍框和所述物體檢測包圍框的交并比，包括：

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

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

6.根據(jù)權利要求1-5中任意一項所述的方法，其特征在于，所述物體真值包圍框的任意邊與所述物體檢測包圍框的任意邊處于非平行狀態(tài)。

7.一種物體檢測模型訓練裝置，其特征在于，包括：

真值模塊：用于根據(jù)檢測目標的標注數(shù)據(jù)確定物體真值包圍框；

檢測模塊：用于將所述檢測目標輸入物體檢測模型，并根據(jù)所述物體檢測模型輸出的檢測數(shù)據(jù)得到物體檢測包圍框；

交并比計算模塊：用于根據(jù)所述物體真值包圍框和所述物體檢測包圍框，確定所述物體真值包圍框和所述物體檢測包圍框的交并比；

損失函數(shù)計算模塊：用于根據(jù)所述交并比，確定損失函數(shù)值；

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

8.根據(jù)權利要求7所述的裝置，其特征在于，在所述物體真值包圍框和所述物體檢測包圍框之間存在交疊的情況下，所述損失函數(shù)計算模塊用于：

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

L＝1-IoU；

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

9.根據(jù)權利要求7所述的裝置，其特征在于，在所述物體真值包圍框和所述物體檢測包圍框之間不存在交疊的情況下，所述損失函數(shù)計算模塊用于：

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

L＝1-GIoU；

其中，

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

10.根據(jù)權利要求7所述的裝置，其特征在于，所述優(yōu)化模塊包括：

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

反向處理單元：用于根據(jù)所述反向傳播計算梯度值，優(yōu)化所述物體檢測模型的參數(shù)。