[發明專利]備選框更新方法及裝置

有效

• 申請號: 201810355286.3
• 申請日: 2018-04-19
• 公開（公告）號: CN110390344B
• 公開（公告）日: 2021-10-26
• 發明（設計）人: 張銳;游驊
• 申請（專利權）人: 華為技術有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62
• 代理公司: 北京東方億思知識產權代理有限責任公司 11258
• 代理人: 賀琳

本申請提供了一種備選框更新方法及裝置，涉及計算機領域。備選框更新方法，包括：根據第一備選框的位置信息和位置向量，得到第一備選框與多個第二備選框的重疊度，第一備選框為備選框集合中分類器得分最高的備選框，第二備選框為備選框集合中除第一備選框之外的備選框，位置向量包括多個第二備選框的位置信息；將第一備選框與多個第二備選框的重疊度和重疊度閾值進行比較以得到更新后的備選框集合。利用本申請的技術方案能夠提高更新備選框的效率，進而提高圖像識別和檢測的工作效率。

技術領域

本申請涉及計算機領域，尤其涉及一種備選框更新的方法及裝置。

背景技術

非極大值抑制(英文：Non-Maximum Suppression，NMS)被廣泛應用在計算機視覺算法中，比如，人臉識別、邊緣檢測或目標檢測等方面。分類器為NMS計算機視覺算法中的重要部分，其可以用于檢測圖像中的對象，比如是否為人臉等，針對圖像中的一個對象，分類器會產生多個備選框。分類器會對每一個備選框計算分類器得分。為了能夠準確地進行對象的檢測或識別，對于一個對象只保留一個最優的備選框，將最優的備選框中的內容作為識別或檢測出的對象。

具體的，選取圖像中所有備選框中分類器得分最高的備選框，遍歷計算其他備選框與分類器得分最高的備選框的重疊面積，根據重疊面積刪除其他備選框中的部分備選框。在未被刪除的其他備選框中再次選取分類器得分最高的備選框，再次執行重疊面積的遍歷計算，直至除最優備選框外的其他備選框全部被刪除。在較復雜的圖像中，備選框的數目非常大，導致遍歷計算量增大，從而使得NMS算法的計算量也隨之增大，降低了圖像識別和檢測的工作效率。

發明內容

本申請提供了一種備選框更新的方法及裝置，能夠提高更新備選框的效率，進而提高圖像識別和檢測的工作效率。

第一方面，本申請提供了一種備選框更新方法，包括：根據第一備選框的位置信息和位置向量，得到第一備選框與多個第二備選框的重疊度，第一備選框為備選框集合中分類器得分最高的備選框，第二備選框為備選框集合中除第一備選框之外的備選框，位置向量包括多個第二備選框的位置信息；將第一備選框與多個第二備選框的重疊度和重疊度閾值進行比較以得到更新后的備選框集合。

在本申請實施例中，利用第一備選框的位置信息和位置向量，通過一次計算即可得到第一備選框與多個第二備選框的重疊度。并可利用計算得到的第一備選框與第二備選框的重疊度與重疊度閾值，得到更新后的備選框集合中的備選框。在更新備選框的過程中，利用位置向量可大大減少更新備選框集合中的備選框所需要的時間，提高了更新備選框的效率，從而提高了利用本申請中備選框更新方法的NMS方法的效率，進而提高了圖像識別和檢測的工作效率。

在第一方面的一種實現方式中，將第一備選框與多個第二備選框的重疊度和重疊度閾值進行比較以得到更新后的備選框集合，包括：在多個第二備選框中的第二備選框和第一備選框的重疊度小于或等于重疊度閾值時，確定第二備選框保留在更新后的備選框集合中；在多個第二備選框中的第二備選框和第一備選框的重疊度大于重疊度閾值時，確定第二備選框不保留在更新后的備選框集合中。

保留與第一備選框的重疊度小于或等于重疊度閾值的第二備選框作為更新后的備選框集合中的備選框，刪除與第一備選框的重疊度大于重疊度閾值的第二備選框。從而減少了參與下一次備選框集合更新的備選框的數目，相應減小了下一次備選框集合更新所需的運算量和占用資源，進一步提高了備選框集合更新的效率。