[發(fā)明專利]目標檢測方法及裝置

有效

• 申請?zhí)枺?/td>: 201711332697.2
• 申請日: 2017-12-13
• 公開（公告）號: CN108009544B
• 公開（公告）日: 2021-08-31
• 發(fā)明（設(shè)計）人: 張水發(fā)
• 申請（專利權(quán)）人: 北京小米移動軟件有限公司
• 主分類號: G06K9/32
• 分類號: G06K9/32
• 代理公司: 北京尚倫律師事務(wù)所 11477
• 代理人: 代治國

1.一種目標檢測方法，其特征在于，包括：

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

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

根據(jù)所述未發(fā)生抑制的檢測框，生成用于確定所述目標圖像內(nèi)目標物的目標檢測框，

所述檢測序列中的檢測框是按照置信度從小到大排列的；對于所述檢測序列中的第i個檢測框；所述i是大于1的整數(shù)；所述根據(jù)所述檢測序列和每個所述檢測框的置信度，保留未發(fā)生抑制的檢測框包括：

獲取第i個檢測框；

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

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

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

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

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

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

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述根據(jù)所述第i個檢測框的置信度和所述重疊比率，確定是否保留所述第i個檢測框包括：

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

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

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

3.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述根據(jù)所述第i個檢測框和所述最低檢測框確定重疊比率包括：

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

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

根據(jù)所述第i個檢測框的坐標范圍和所述最低檢測框的坐標范圍，確定兩者的交集面積和并集面積；

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

4.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述從目標圖像中提取由多個檢測框組成的檢測序列及每個所述檢測框的置信度包括：

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

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

5.一種目標檢測裝置，其特征在于，包括：

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

保留模塊，用于根據(jù)所述檢測序列和每個所述檢測框的置信度，保留未發(fā)生抑制的檢測框；

生成模塊，用于根據(jù)所述未發(fā)生抑制的檢測框，生成用于確定所述目標圖像內(nèi)目標物的目標檢測框，

所述檢測序列中的檢測框是按照置信度從小到大排列的；對于所述檢測序列中的第i個檢測框；所述i是大于1的整數(shù)；所述保留模塊包括：

第一獲取子模塊，用于獲取第i個檢測框；

第二獲取子模塊，用于從已保留的檢測框中獲取置信度最低的最低檢測框；

第一確定子模塊，用于根據(jù)所述第i個檢測框和所述最低檢測框確定重疊比率；

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

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

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

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

6.根據(jù)權(quán)利要求5所述的裝置，其特征在于，所述第二確定子模塊包括：

第一確定單元，用于根據(jù)所述第i個檢測框的置信度和所述重疊比率確定所述第i個檢測框的新置信度；

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

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