[發明專利]目標檢測方法及裝置

有效

• 申請號: 201711332697.2
• 申請日: 2017-12-13
• 公開（公告）號: CN108009544B
• 公開（公告）日: 2021-08-31
• 發明（設計）人: 張水發
• 申請（專利權）人: 北京小米移動軟件有限公司
• 主分類號: G06K9/32
• 分類號: G06K9/32
• 代理公司: 北京尚倫律師事務所 11477
• 代理人: 代治國

本公開是關于目標檢測方法及裝置。該方法包括：從目標圖像中提取由多個檢測框組成的檢測序列及每個所述檢測框的置信度；根據所述檢測序列和每個所述檢測框的置信度，保留未發生抑制的檢測框；根據所述未發生抑制的檢測框，生成用于確定所述目標圖像內目標物的目標檢測框。該技術方案可以根據置信度到對可能發生抑制的檢測框進行再判斷，能夠在某些誤檢的情況下找回一些檢測框，提高檢測的準確性。

技術領域

本公開涉及圖像處理領域，尤其涉及目標檢測方法及裝置。

背景技術

非極大值抑制在目標檢測、目標定位等領域是一種被廣泛使用的方法。對于目標具體位置定位過程，不管是基于滑動窗口還是選擇性搜索方法，都會產生好多的候選區域。實際看到的情形就是好多區域的交叉重疊，難以滿足實際的應用。目前的非極大值抑制方法會刪除所有的發生抑制的候選區域。

發明內容

本公開實施例提供目標檢測方法及裝置。所述技術方案如下：

根據本公開實施例的第一方面，提供一種目標檢測方法，包括：

從目標圖像中提取由多個檢測框組成的檢測序列及每個所述檢測框的置信度；

根據所述檢測序列和每個所述檢測框的置信度，保留未發生抑制的檢測框；

根據所述未發生抑制的檢測框，生成用于確定所述目標圖像內目標物的目標檢測框。

在一個實施例中，所述檢測序列中的檢測框是按照置信度從小到大排列的；對于所述檢測序列中的第i個檢測框；所述i是大于1的整數；所述根據所述檢測序列和每個所述檢測框的置信度，保留未發生抑制的檢測框包括：

獲取第i個檢測框；

從已保留的檢測框中獲取置信度最低的最低檢測框；

根據所述第i個檢測框和所述最低檢測框確定重疊比率；

當所述重疊比率大于或等于第一閾值時，刪除所述第i個檢測框；

當所述重疊比率小于所述第一閾值，且大于或等于第二閾值時，根據所述第i個檢測框的置信度和所述重疊比率，確定是否保留所述第i個檢測框；

當所述重疊比率小于所述第二閾值時，保留所述第i個檢測框；

其中，所述未發生抑制的檢測框包括所有已保留的檢測框。

在一個實施例中，所述根據所述第i個檢測框的置信度和所述重疊比率，確定是否保留所述第i個檢測框包括：

根據所述第i個檢測框的置信度和所述重疊比率確定所述第i個檢測框的新置信度；

判斷所述新置信度是否大于或等于第三閾值；

其中，當所述新置信度大于或等于第三閾值時，保留所述第i個檢測框；當所述新置信度小于所述第三閾值時，刪除所述第i個檢測框。

在一個實施例中，所述根據所述第i個檢測框和所述最低檢測框確定重疊比率包括：

獲取所述第i個檢測框的坐標范圍；

獲取所述最低檢測框的坐標范圍；

根據所述第i個檢測框的坐標范圍和所述第一個檢測框的坐標范圍，確定兩者的交集面積和并集面積；

根據所述交集面積和所述并集面積，確定所述重疊比率。

在一個實施例中，所述從目標圖像中提取由多個檢測框組成的檢測序列及每個所述檢測框的置信度包括：

從目標圖像中提取多個檢測框及每個所述檢測框的置信度；

將所述多個檢測框按置信度從大到小排序，得到所述檢測序列。