[發明專利]物體檢測方法、裝置、設備及介質

在審

• 申請號: 202110565334.3
• 申請日: 2021-05-24
• 公開（公告）號: CN113379683A
• 公開（公告）日: 2021-09-10
• 發明（設計）人: 商明陽;向大衛;王志成
• 申請（專利權）人: 北京邁格威科技有限公司
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06T7/70
• 代理公司: 北京潤澤恒知識產權代理有限公司 11319
• 代理人: 王婷婷

本發明實施例提供了一種物體檢測方法、裝置、設備及介質，屬于圖像處理技術領域，旨在提高物體檢測的準確性，所述方法包括：對待檢測圖像中的物體的局部區域進行檢測，得到所述物體的至少一個候選局部框，其中，一個物體的候選局部框用于框選該物體的局部區域；從所述物體的至少一個候選局部框中，篩選出所述物體的目標局部框；根據所述物體的目標局部框，確定所述物體的完整框，其中，一個物體的完整框表征該物體的檢測結果。

技術領域

本發明涉及圖像處理技術領域，特別是涉及一種物體檢測方法、裝置、設備及介質。

背景技術

物體檢測是利用計算機視覺技術確定圖片和視頻中的物體位置的技術。其中，在進行物體檢測時，一般是先確定出物體的多個候選框，之后，對多個候選框去重后保留一個候選框作為物體的目標框。該目標框即框選出了物體在圖像中的位置，后續便可以對目標框中的物體進行識別和處理。

然而，在一張圖像中往往存在多個需要待檢測的物體，此種情況下，便需要框選出多個物體的目標框。此種情況下，往往在對多個物體得到的多個候選框進行去重時，會使得一些相互有重疊的物體的目標框被漏檢。例如，以物體檢測中的行人檢測為例，對于密集的行人檢測，行人A和行人B具有遮擋關系，此時對行人A和行人B進行檢測時，可能將行人B的候選框全部去除，而只保留了行人A的候選框，使得最終確定出的行人的數量不準確。

綜上，現有的物體檢測會導致物體檢測的準確率不高的問題。

發明內容

鑒于上述問題，提出了本發明實施例的一種物體檢測方法、裝置、設備及介質，以便克服上述問題或者至少部分地解決上述問題。

為了解決上述問題，本發明的第一方面，公開了一種物體檢測方法，所述方法包括：

對待檢測圖像中的物體的局部區域進行檢測，得到所述物體的至少一個候選局部框，其中，一個物體的候選局部框用于框選該物體的局部區域；

從所述物體的至少一個候選局部框中，篩選出所述物體的目標局部框；

根據所述物體的目標局部框，確定所述物體的完整框，其中，一個物體的完整框表征該物體的檢測結果。

可選地，從所述物體的至少一個候選局部框中篩選出一個候選局部框，作為所述物體的目標局部框，包括：

根據所述物體的至少一個候選局部框之間的重疊度，從所述物體的至少一個候選局部框中篩選出一個候選局部框，作為所述物體的目標局部框；或

根據所述物體的至少一個候選局部框分別與預先標記的所述物體的中心之間的距離，從所述物體的至少一個候選局部框中篩選出一個候選局部框，作為所述物體的目標局部框。

可選地，根據所述物體的目標局部框，確定所述物體的完整框，包括：

根據所述物體的目標局部框和所述目標局部框對應的局部特征圖，確定所述物體的目標局部結構；

根據所述物體的完整結構與所述物體的各個局部結構之間的相對位置關系和所述物體的目標局部結構，確定所述物體的完整框。

可選地，根據所述物體的目標局部框，確定所述物體的完整框，包括：

將所述物體的目標局部框和所述目標局部框所框選的圖像區域的特征圖，輸入預先訓練的第一物體檢測模型，得到所述物體的完整框，其中，所述第一物體檢測模型是利用預先標記有樣本物體的局部框和完整框的樣本圖像訓練得到的；或

將所述物體的目標局部框和所述待檢測圖像的特征圖，輸入預先訓練的第二物體檢測模型，得到所述物體的完整框，其中，所述第二物體檢測模型是利用預先標記有樣本物體的局部框和完整框的樣本圖像訓練得到的。

可選地，對待檢測圖像中的物體的局部區域進行檢測，得到所述物體的至少一個候選局部框，包括：