[發明專利]檢測框標定方法、裝置、設備及計算機可讀存儲介質

在審

• 申請號: 202210033134.8
• 申請日: 2022-01-12
• 公開（公告）號: CN116486063A
• 公開（公告）日: 2023-07-25
• 發明（設計）人: 魏濤;劉陽興
• 申請（專利權）人: 武漢TCL集團工業研究院有限公司
• 主分類號: G06V10/24
• 分類號: G06V10/24;G06V10/774
• 代理公司: 深圳紫藤知識產權代理有限公司 44570
• 代理人: 韓金濤

1.一種檢測框標定方法，其特征在于，包括：

獲取待檢測的目標圖像；

根據所述目標圖像生成網格圖像；

將所述網格圖像輸入已訓練的檢測模型，得到所述網格圖像對應的檢測框偏移量；

根據所述網格圖像在所述目標圖像中的位置信息，及其對應的檢測框偏移量，確定目標檢測框。

2.根據權利要求1所述的方法，其特征在于，所述根據所述網格圖像在所述目標圖像中的位置信息，及其對應的檢測框偏移量，確定目標檢測框，包括：

根據各所述網格圖像在所述目標圖像中的位置信息，及其對應的檢測框偏移量，確定各所述網格圖像對應的檢測框中心點；

根據各所述網格圖像對應的檢測框中心點之間的距離，對所述檢測框中心點進行劃分，得到目標檢測框中心點集合；

根據所述目標檢測框中心點集合中的檢測框中心點，生成目標檢測框。

3.根據權利要求2所述的方法，其特征在于，所述根據所述目標檢測框中心點集合中的檢測框中心點，生成目標檢測框，包括：

計算所述目標檢測框中心點集合中各檢測框中心點的中心位置，得到目標中心點；

根據所述目標中心點的位置信息以及預設的目標檢測框尺寸，得到目標角點；

根據所述目標角點生成目標檢測框。

4.根據權利要求3所述的方法，其特征在于，所述根據所述目標中心點的位置信息以及預設的目標檢測框尺寸，得到目標角點之前，所述方法還包括：

計算所述網格圖像對應的檢測框尺寸的平均值，得到目標檢測框尺寸，所述網格圖像對應的檢測框尺寸是通過將所述網格圖像輸入所述已訓練的檢測模型得到。

5.根據權利要求2所述的方法，其特征在于，所述目標檢測框中心點集合中各檢測框中心點之間的距離小于預設的距離閾值。

6.根據權利要求1所述的方法，其特征在于，所述將所述網格圖像輸入已訓練的檢測模型，得到所述網格圖像對應的檢測框偏移量之前，所述方法還包括：

獲取樣本圖像，以及所述樣本圖像對應的樣本網格圖像和標簽檢測框；

根據各所述樣本網格圖像在所述樣本圖像中的位置信息，以及所述樣本圖像對應的標簽檢測框的位置信息，設定各所述樣本網格圖像對應的檢測框標簽偏移量；

根據各所述樣本網格圖像及其對應的檢測框標簽偏移量，對預設的初始模型進行訓練，得到已訓練的檢測模型。

7.根據權利要求6所述的方法，其特征在于，所述根據各所述樣本網格圖像及其對應的檢測框標簽偏移量，對預設的初始模型進行訓練，得到已訓練的檢測模型，包括：

將所述樣本網格圖像輸入預設的初始模型，得到所述樣本網格圖像對應的檢測框預測偏移量；

根據所述樣本網格圖像在所述樣本網格圖像中的位置信息，及其對應的檢測框預測偏移量，確定預測檢測框；

根據所述檢測框預測偏移量和所述檢測框標簽偏移量的差值，以及所述預測檢測框和所述標簽檢測框的位置關系，確定所述初始模型對應的損失值；

根據所述損失值對所述初始模型進行訓練，得到已訓練的檢測模型。

8.根據權利要求7所述的方法，其特征在于，所述根據所述檢測框預測偏移量和所述檢測框標簽偏移量的差值，以及所述預測檢測框和所述標簽檢測框的位置關系，確定所述初始模型對應的損失值，包括：

根據所述檢測框預測偏移量和所述檢測框標簽偏移量的差值，得到第一損失值；

根據所述預測檢測框和所述標簽檢測框的交集和并集，得到第一檢測框和第二檢測框；

根據所述第一檢測框和所述第二檢測框的面積比，得到第二損失值；

根據所述第一損失值和所述第二損失值，得到所述初始模型對應的損失值。