[發明專利]目標檢測中的真值框選擇方法、裝置、存儲介質及設備

在審

• 申請號: 202011135858.0
• 申請日: 2020-10-22
• 公開（公告）號: CN112257586A
• 公開（公告）日: 2021-01-22
• 發明（設計）人: 王堃
• 申請（專利權）人: 江蘇禹空間科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00
• 代理公司: 北京德崇智捷知識產權代理有限公司 11467
• 代理人: 黃雪

1.一種目標檢測中的真值框選擇方法，其特征在于，所述方法包括：

獲取對圖像進行特征提取后得到的目標特征圖，所述目標特征圖中包含多個預定大小的網格；

在所述目標特征圖中獲取所述圖像中的每個小目標對象對應的多個檢測框，所述小目標對象是像素面積小于預定面積的目標對象；

對于每個檢測框，計算預定點位于所述檢測框內的網格的中心度得分，所述預定點為所述網格的角點和/或中心點；

對于每個小目標對象，從所述小目標對象對應的多個檢測框中，將最大的中心度得分對應的檢測框確定為所述小目標對象的真值框。

2.根據權利要求1所述的方法，其特征在于，所述對于每個檢測框，計算預定點位于所述檢測框內的網格的中心度得分，包括：

對于每個檢測框，獲取位于所述檢測框內的每個預定點的坐標信息；

對每個預定點的坐標信息進行回歸運算，得到每個預定點的回歸結果；

對每個預定點的回歸結果進行中心度運算，得到每個預定點的中心度得分；

將所有預定點的中心度得分相加，得到所述檢測框對應的中心度得分。

3.根據權利要求2所述的方法，其特征在于，所述對每個預定點的坐標信息進行回歸運算，得到每個預定點的回歸結果，包括：

對于每個預定點，根據所述預定點的坐標信息，計算所述預定點到所述檢測框的左邊框的第一距離，計算所述預定點到所述檢測框的上邊框的第二距離，計算所述預定點到所述檢測框的有邊框的第三距離，計算所述預定點到所述檢測框的下邊框的第四距離；

將所述預定點的橫坐標減去所述第一距離，得到第一結果；

將所述預定點的縱坐標減去所述第二距離，得到第二結果；

將所述預定點的橫坐標加上所述第三距離，得到第三結果；

將所述預定點的縱坐標加上所述第四距離，得到第四結果；

將所述第一結果、所述第二結果、所述第三結果和所述第四結果確定為所述預定點的回歸結果。

4.根據權利要求3所述的方法，其特征在于，所述對每個預定點的回歸結果進行中心度運算，得到每個預定點的中心度得分，包括：

對于每個預定點，將第一數值除以第二數值，得到第一比值，所述第一數值是所述預定點對應的所述第一結果和所述第三結果中的最小值，所述第二數值是所述第一結果和所述第三結果中的最大值；

將第三數值除以第四數值，得到第二比值，所述第三數值是所述預定點對應的所述第二結果和所述第四結果中的最小值，所述第四數值是所述第二結果和所述第四結果中的最大值；

對所述第一比值和所述第二比值的乘積進行開方運算，得到所述預定點的中心度得分。

5.根據權利要求2所述的方法，其特征在于，所述獲取位于所述檢測框內的每個預定點的坐標信息，包括：

以所述檢測框的中心為原點創建坐標系；

計算位于所述檢測框內的每個預定點在所述坐標系中的坐標信息。

6.根據權利要求1至5中任一項所述的方法，其特征在于，所述獲取對圖像進行特征提取后得到的目標特征圖，包括：

對所述圖像進行特征提取，得到多個特征圖；

從所述多個特征圖中選擇用于檢測所述小目標對象的目標特征圖。

7.根據權利要求6所述的方法，其特征在于，所述從所述多個特征圖中選擇用于檢測所述小目標對象的目標特征圖，包括：

利用特征金字塔網絡FPN對所述多個特征圖進行運算，得到多個預測層；

將所述多個預測層中的最頂層預測層確定為用于檢測所述小目標對象的目標特征圖。