[發明專利]一種基于軟標簽的目標檢測樣本平衡方法

有效

• 申請號: 202110358650.3
• 申請日: 2021-04-01
• 公開（公告）號: CN113128575B
• 公開（公告）日: 2022-03-11
• 發明（設計）人: 謝雪梅;王振愿;李錦航
• 申請（專利權）人: 西安電子科技大學廣州研究院
• 主分類號: G06V10/774
• 分類號: G06V10/774;G06V10/25;G06K9/62
• 代理公司: 廣州三環專利商標代理有限公司 44202
• 代理人: 麥小嬋;郝傳鑫

1.一種基于軟標簽的目標檢測樣本平衡方法，其特征在于，包括：

獲取輸入圖像的目標的個數以及候選框的個數；

計算每一個目標與每一個候選框的第一IOU值；

對所述第一IOU值按列取最大值，得到第二IOU值，并獲取所述第二IOU值在所述第一IOU值中的行索引值；

對每一個目標，從所述行索引值中獲取負責該目標的所有目標候選框，基于軟標簽獲取每一個目標候選框的標簽值；

根據每一個目標候選框的標簽值以及預先構建的損失函數獲取每一個目標候選框的預測概率；

根據所述預測概率輸出目標檢測框；

其中，所述基于軟標簽獲取每一個目標候選框的標簽值，具體為：

根據如下公式計算每一個目標候選框的標簽值：

式中，MIou(A)為負責該目標的所有目標候選框的第二IOU值，MIou(a)為負責該目標的目標候選框a的第二IOU值，w為負責該目標的目標候選框a的標簽值。

2.根據權利要求1所述的基于軟標簽的目標檢測樣本平衡方法，其特征在于，所述根據每一個目標候選框的標簽值以及預先構建的損失函數獲取每一個目標候選框的預測概率，具體為：

將每一個目標候選框的標簽值輸入所述預先構建的第一損失函數；所述預先構建的第一損失函數的表達式為：Sloss＝-(wlog(p_x)+(1-w)log(1-p_x))-bias，其中，bias＝-(wlog(w)+(1-w)log(1-w))，p_x為第一預測概率；

將令所述預先構建的第一損失函數損失最小的第一預測概率作為所述預測概率。

3.根據權利要求1所述的基于軟標簽的目標檢測樣本平衡方法，其特征在于，所述根據每一個目標候選框的標簽值以及預先構建的損失函數獲取每一個目標候選框的預測概率，具體為：

將每一個目標候選框的標簽值輸入所述預先構建的第二損失函數；其中，所述預先構建的第二損失函數的表達式為：SFloss＝(w-p′_x)^γ(-(wlog(p′_x)+(1-w)log(1-p′_x))-bias)，bias＝-(wlog(w)+(1-w)log(1-w))，p′_x為第二預測概率，γ為一個超參數；

將令所述預先構建的第二損失函數損失最小的第二預測概率作為所述預測概率。

4.根據權利要求1所述的基于軟標簽的目標檢測樣本平衡方法，其特征在于，所述根據所述預測概率輸出目標檢測框，具體包括：

去除所述預測概率小于等于預設閾值的目標候選框；

通過非極大值抑制算法去除剩余目標候選框中重復的冗余目標候選框，得到所述目標檢測框；

輸出所述目標檢測框。

5.根據權利要求4所述的基于軟標簽的目標檢測樣本平衡方法，其特征在于，所述預設閾值為0.05。

6.根據權利要求1所述的基于軟標簽的目標檢測樣本平衡方法，其特征在于，在獲取輸入圖像的目標的個數以及候選框的個數，所述方法還包括：

獲取原始圖像，對所述原始圖像進行預處理，得到所述輸入圖像。