[發明專利]圖像過濾方法、裝置、電子設備及存儲介質

在審

• 申請號: 202011642690.2
• 申請日: 2020-12-30
• 公開（公告）號: CN113158733A
• 公開（公告）日: 2021-07-23
• 發明（設計）人: 陳浩彬;劉凱鑒;余世杰;陳大鵬;趙瑞
• 申請（專利權）人: 北京市商湯科技開發有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/32;G06K9/62;G06N3/04
• 代理公司: 廣州三環專利商標代理有限公司 44202
• 代理人: 熊永強;董文俊

1.一種圖像過濾方法，其特征在于，包括：

對待過濾圖像進行人體檢測，得到N個人體框和N個人體掩碼圖，其中，所述N個人體框和所述N個人體掩碼圖一一對應，N為大于或等于1的整數；

根據所述N個人體框和所述N個人體掩碼圖對所述待過濾圖像進行過濾。

2.根據權利要求1所述的方法，其特征在于，所述根據所述N個人體框和所述N個人體掩碼圖對所述待過濾圖像進行過濾，包括：

獲取所述N個人體框中每個人體框的置信度；

根據所述每個人體框的置信度以及所述每個人體框對應的人體掩碼圖，對所述N個人體框進行非極大值抑制，得到K個人體框，K為小于或等于N的整數；

根據所述K個人體框以及與所述K個人體框對應的K人體掩碼圖，對所述待過濾圖像進行過濾。

3.根據權利要求2所述的方法，其特征在于，所述根據所述每個人體框的置信度以及所述每個人體框對應的人體掩碼圖，對所述N個人體框進行非極大值抑制，得到K個人體框，包括：

根據所述每個人體框的置信度對所述N個人體框進行降序排序，得到第一集合；

將第一人體框從所述第一集合移出，得到第二集合，所述第一人體框為所述N個人體框中置信度最大的人體框；

確定第一人體掩碼圖和第二人體掩碼圖之間的第一交并比，所述第一人體掩碼圖為與所述第一人體框對應的人體掩碼圖，所述第二人體掩碼圖為與第二人體框對應的人體掩碼圖，所述第二人體框為所述第二集合中的任意一個人體框；

在所述第一交并比大于第一閾值的情況下，將所述第二人體框從所述第二集合中刪除，得到第三集合，在所述第一交并比小于或等于所述第一閾值的情況下，將所述第二人體框保留在所述第二集合中，并將所述第二集合作為第三集合；

將所述第三集合作為新的第一集合，并對所述新的第一集合的人體框進行非極大值抑制，在所述新的第一集合中的元素為零的情況下，將對所述N個人體框進行非極大值抑制過程中移出的人體框作為所述K個人體框。

4.根據權利要求3所述的方法，其特征在于，所述確定第一人體掩碼圖和第二人體掩碼圖之間的第一交并比，包括：

對所述第一人體掩碼圖和所述第二人體掩碼圖之間的相交區域中的各個像素點的像素值求和，得到第一掩碼面積；

對所述第一人體掩碼圖中的各個像素點的像素值求和，得到第二掩碼面積；

對所述第二人體掩碼圖中的各個像素點的像素值求和，得到第三掩碼面積；

根據所述第一掩碼面積、所述第二掩碼面積以及所述第三掩碼面積，確定所述第一人體掩碼圖和所述第二人體掩碼圖之間的第一交并比。

5.根據權利要求4所述的方法，其特征在于，所述根據所述第一掩碼面積、所述第二掩碼面積以及所述第三掩碼面積，確定所述第一人體掩碼圖和所述第二人體掩碼圖之間的第一交并比，包括：

確定所述第一掩碼面積相對于所述第二掩碼面積的第一比值；

確定所述第一掩碼面積相對于所述第三掩碼面積的第二比值；

將所述第一比值和所述第二比值中的最大比值作為所述第一交并比。

6.根據權利要求1-5中任一項所述的方法，其特征在于，所述根據所述N個人體框和所述N個人體掩碼圖對所述待過濾圖像進行過濾，包括：

對所述K個人體掩碼圖中的每個人體掩碼圖中的各個像素點的像素值求和，得到所述K個人體掩碼圖中的每個人體掩碼圖對應的掩碼面積；

確定第三人體掩碼圖的掩碼面積與第四人體掩碼圖的掩碼面積的第三比值，其中，所述第三人體掩碼圖為所述K個人體掩碼圖中掩碼面積最大的人體掩碼圖，所述第四人體掩碼圖為所述K個人體掩碼圖中掩碼面積僅次于所述第三人體掩碼圖的人體掩碼圖；

在所述第三比值大于或等于第二閾值的情況下，保留所述待過濾圖像，在所述第三比值小于所述第二閾值的情況下，濾除所述待過濾圖像。

7.根據權利要求6所述的方法，其特征在于，在保留所述待過濾圖像的情況下，所述方法還包括：

對所述待過濾圖像進行人臉檢測，得到M個人臉掩碼圖，M為大于或等于0的整數；

根據所述第三人體掩碼圖以及所述M個人臉掩碼圖，構造訓練樣本。