[發明專利]一種檢測模型的處理方法及裝置

有效

• 申請號: 202011506742.3
• 申請日: 2020-12-18
• 公開（公告）號: CN112633496B
• 公開（公告）日: 2023-08-08
• 發明（設計）人: 陳偉杰;馬振鑫;陳斌斌;過一路;趙龍;楊世才;謝迪;浦世亮
• 申請（專利權）人: 杭州海康威視數字技術股份有限公司
• 主分類號: G06N3/088
• 分類號: G06N3/088
• 代理公司: 北京中博世達專利商標代理有限公司 11274
• 代理人: 孫靜

1.一種檢測模型的處理方法，其特征在于，所述方法包括：

獲取N幀圖片；所述N幀圖片為在線采集的圖片，N為正整數；

將所述N幀圖片輸入到第一檢測模型，進行對象檢測處理，得到每幀圖片中檢測框的置信度；所述檢測框用于表征圖片中對象的位置；第一檢測模型為初始檢測模型或者為當前的檢測模型；

根據第一置信度閾值和獲取到的檢測框的置信度，確定正樣本和負樣本；所述正樣本包括第一檢測框和所述N幀圖片中包括所述第一檢測框的圖片，所述負樣本包括第二檢測框和所述N幀圖片中包括所述第二檢測框的圖片；所述第一檢測框為置信度高于所述第一置信度閾值的檢測框，所述第二檢測框為置信度低于所述第一置信度閾值的檢測框；所述第一置信度閾值大于所述第一檢測模型對應的置信度閾值；

根據所述正樣本和所述負樣本對所述第一檢測模型進行訓練，得到第二檢測模型；

獲取M幀圖片；所述M幀圖片為在線采集的圖片，所述M幀圖片與所述N幀圖片不同，M為正整數；

將所述M幀圖片輸入到所述第一檢測模型，進行對象檢測處理，得到每幀圖片中檢測框的第一置信度；

將所述M幀圖片輸入到所述第二檢測模型，進行對象檢測處理，得到每幀圖片中檢測框的第二置信度；

確定第一數量和第二數量；所述第一數量為第一置信度高于所述第一檢測模型對應的置信度閾值的檢測框的數量；所述第二數量為第二置信度高于所述第一檢測模型對應的置信度閾值的檢測框的數量；所述第一數量用于表征所述第一檢測模型的檢測能力，所述第二數量用于表征所述第二檢測模型的檢測能力；

若所述第一數量大于所述第二數量，則確定所述第一檢測模型的檢測能力優于所述第二檢測模型的檢測能力；

在確定所述第二檢測模型的檢測能力優于所述第一檢測模型的檢測能力的情況下，確定所述第二檢測模型對應的置信度閾值，所述第二檢測模型用于對后續采集到的圖片的檢測；

在確定所述第一檢測模型的檢測能力優于所述第二檢測模型的檢測能力的情況下，重新確定正樣本和負樣本；根據重新確定的正樣本和負樣本，對所述第一檢測模型進行訓練。

2.根據權利要求1所述的方法，其特征在于，所述在確定所述第一檢測模型的檢測能力優于所述第二檢測模型的檢測能力的情況下，重新確定正樣本和負樣本，包括：

確定所述重新確定的正樣本包括所述負樣本中滿足預設條件的樣本和所述正樣本；

確定所述重新確定的負樣本包括所述負樣本中除滿足所述預設條件的樣本以外的其他樣本。

3.根據權利要求1所述的方法，其特征在于，所述方法還包括：

獲取在線采集到的初始圖片；

對所述初始圖片去冗余處理，得到目標圖片；所述目標圖片包括所述N幀圖片和所述M幀圖片。

4.根據權利要求3所述的方法，其特征在于，所述對所述初始圖片去冗余處理，得到目標圖片，包括：

將所述初始圖片輸入到所述當前的檢測模型，進行對象檢測處理，得到每幀所述初始圖片中檢測框；

對每幀所述初始圖片執行第一操作，以得到目標圖片；其中，所述第一操作包括：若第i幀初始圖片中檢測框與對應的第i-1幀初始圖片中檢測框之間的重疊度大于預設重疊度閾值，則刪除所述第i幀初始圖片；i∈[1，K]，K等于所述初始圖片的數量。

5.根據權利要求3所述的方法，其特征在于，所述對所述初始圖片去冗余處理，得到目標圖片，包括：

將所述初始圖片輸入到所述當前的檢測模型，進行對象檢測處理，得到每幀所述初始圖片中檢測框、以及每幀所述初始圖片對應的中間特征圖；

對每幀所述初始圖片執行第一操作，以得到待選圖片；其中，所述第一操作包括：若第i幀初始圖片中檢測框與對應的第i-1幀初始圖片中檢測框之間的重疊度大于預設重疊度閾值，則刪除所述第i幀初始圖片；i∈[1，K]，K等于所述初始圖片的數量；

在所述待選圖片的數量大于第一預設閾值的情況下，根據所述待選圖片和所述待選圖片對應的中間特征圖確定所述待選圖片中檢測框的特征，并根據所述待選圖片中檢測框的特征，對所述待選圖片去冗余處理，得到所述目標圖片。