[發明專利]一種目標檢測模型的訓練方法及裝置

在審

• 申請號: 201910275732.4
• 申請日: 2019-04-08
• 公開（公告）號: CN110135456A
• 公開（公告）日: 2019-08-16
• 發明（設計）人: 李誠;張險峰
• 申請（專利權）人: 圖麟信息科技（上海）有限公司;合肥麟圖信息科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62
• 代理公司: 北京柏杉松知識產權代理事務所(普通合伙) 11413
• 代理人: 李欣;項京

本申請實施例提供了一種目標檢測模型的訓練方法及裝置，涉及圖像處理技術領域。該方法包括：通過待訓練的目標檢測模型確定第一樣本圖像中包含檢測對象的檢測區域，根據第一樣本圖像中預先標記的包含檢測對象的目標區域，確定檢測區域與目標區域之間的重疊度，并將重疊度小于預設第一閾值的檢測區域標記為背景區域，根據第一樣本圖像中標記的背景區域和標記的目標區域，對待訓練的目標檢測模型進行訓練，將訓練得到的目標檢測模型作為當前的待訓練的目標檢測模型，并返回通過待訓練的目標檢測模型，確定第一樣本圖像中包含檢測對象的檢測區域的步驟，以使訓練得到的目標檢測模型滿足預設準確度條件。采用本申請可以降低目標檢測模型的誤檢率。

技術領域

本申請涉及圖像處理技術領域，特別是涉及一種目標檢測模型的訓練方法及裝置。

背景技術

目標檢測是檢測圖像中檢測對象的位置的一種技術，在人臉識別、車輛識別以及動態追蹤等方面都有廣泛的應用前景。目標檢測是通過目標檢測模型檢測圖像中檢測對象(例如車輛)的位置信息，例如，輸出包圍檢測對象的邊界框位置信息?，F有技術中，誤檢(即將圖像中背景區域檢測成目標區域)是影響目標檢測模型應用的一個重要問題。

為了降低目標檢測模型的誤檢率，現有技術中在根據樣本圖像中標記的檢測對象的目標區域對目標檢測模型訓練的情況下，還會在樣本圖像中除檢測對象的目標區域以外的區域中，隨機標記背景區域，然后，根據樣本圖像中的標記的背景區域，對目標檢測模型進行訓練。

然而，現有技術中樣本圖像中標記的背景區域是隨機標記的，根據樣本圖像中隨機標記的背景區域，對目標檢測模型進行訓練，訓練后的目標檢測模型的誤檢率較高。

發明內容

本申請實施例的目的在于提供一種目標檢測模型的訓練方法及裝置，可以降低目標檢測模型的誤檢率。具體技術方案如下：

第一方面，提供了一種目標檢測模型的訓練方法，所述方法包括：

通過待訓練的目標檢測模型，確定第一樣本圖像中包含檢測對象的檢測區域；

根據所述第一樣本圖像中預先標記的包含所述檢測對象的目標區域，確定所述檢測區域與所述目標區域之間的重疊度；

將重疊度小于預設第一閾值的檢測區域，標記為背景區域；

根據所述第一樣本圖像中標記的背景區域和標記的目標區域，對所述待訓練的目標檢測模型進行訓練，將訓練得到的目標檢測模型作為當前的待訓練的目標檢測模型，并返回所述通過待訓練的目標檢測模型，確定第一樣本圖像中包含檢測對象的檢測區域的步驟，以使訓練得到的目標檢測模型滿足預設準確度條件。

可選的，所述通過待訓練的目標檢測模型，確定第一樣本圖像中包含檢測對象的檢測區域之前，所述方法還包括：

根據標記有包含檢測對象的目標區域的初始樣本圖像，對初始的目標檢測模型進行訓練，直至訓練后的目標檢測模型的損失函數收斂；

將訓練后的目標檢測模型作為待訓練的目標檢測模型。

可選的，所述根據所述第一樣本圖像中預先標記的包含所述檢測對象的目標區域，確定所述檢測區域與所述目標區域之間的重疊度，包括：

根據所述第一樣本圖像中預先標記的包含所述檢測對象的目標區域，確定所述檢測區域與所述目標區域的交集面積，及所述檢測區域與所述目標區域的并集面積；

確定所述交集面積與所述并集面積的比值，得到所述檢測區域與所述目標區域之間的重疊度。

可選的，所述根據所述第一樣本圖像中標記的背景區域和標記的目標區域，對所述待訓練的目標檢測模型進行訓練，包括：

根據所述第一樣本圖像中標記的目標區域的個數和預設比率，確定背景區域的需求個數；