[發明專利]遮擋物識別模型訓練方法、識別方法、裝置及存儲介質

在審

• 申請號: 202010334114.5
• 申請日: 2020-04-24
• 公開（公告）號: CN111860566A
• 公開（公告）日: 2020-10-30
• 發明（設計）人: 張修寶;沈海峰
• 申請（專利權）人: 北京嘀嘀無限科技發展有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06K9/00
• 代理公司: 北京超成律師事務所 11646
• 代理人: 裴素英

1.一種遮擋物識別模型訓練方法，其特征在于，包括：

獲取人臉樣本集合、遮擋物負樣本集合，其中，所述人臉樣本集合包括：多個未遮擋人臉圖像；所述遮擋物負樣本集合包括：多種非目標遮擋物的圖像以及所述非目標遮擋物的標簽；

根據所述人臉樣本集合、所述遮擋物負樣本集合以及預設組合方式，組合獲取人臉遮擋樣本，其中，所述人臉遮擋樣本包括：不同所述非目標遮擋物遮擋后的人臉圖像以及所述非目標遮擋物的標簽；

采用所述人臉遮擋樣本和預設深度學習網絡，訓練獲取遮擋物識別模型。

2.根據權利要求1所述的方法，其特征在于，所述獲取人臉樣本集合，包括：

獲取包含未遮擋人臉的原始圖像；

采用人臉檢測算法檢測獲取所述原始圖像中的人臉；

采用預設關鍵點檢測算法檢測并標注所述人臉的多個關鍵點坐標，將標注后的人臉圖像加入所述人臉樣本集合。

3.根據權利要求2所述的方法，其特征在于，所述根據所述人臉樣本集合、所述遮擋物負樣本集合以及預設組合方式，組合獲取人臉遮擋樣本，包括：

在所述人臉樣本集合中隨機選取一個待組合人臉圖像、在所述遮擋物負樣本集合中隨機選取一個待組合非目標遮擋物的圖像；

根據所述待組合人臉圖像上的關鍵點坐標、以及所述預設組合方式，將所述待組合人臉圖像與所述待組合非目標遮擋物的圖像進行組合，獲取所述非目標遮擋物遮擋后的人臉圖像；

將所述非目標遮擋物遮擋后的人臉圖像加入所述人臉遮擋樣本。

4.根據權利要求3所述的方法，其特征在于，所述預設組合方式包括多種組合方式，不同組合方式基于所述關鍵點坐標的遮擋位置、遮擋范圍不同。

5.根據權利要求4所述的方法，其特征在于，所述根據所述待組合人臉圖像上的關鍵點坐標、以及所述預設組合方式，將所述待組合人臉圖像與所述待組合非目標遮擋物的圖像進行組合，獲取所述非目標遮擋物遮擋后的人臉圖像，包括：

在多種所述組合方式中隨機選取一種目標組合方式；

根據所述待組合人臉圖像上的關鍵點坐標、以及所述目標組合方式，將所述待組合人臉圖像與所述待組合非目標遮擋物的圖像進行組合，獲取所述非目標遮擋物遮擋后的人臉圖像。

6.根據權利要求5所述的方法，其特征在于，所述根據所述待組合人臉圖像上的關鍵點坐標、以及所述目標組合方式，將所述待組合人臉圖像與所述待組合非目標遮擋物的圖像進行組合，獲取所述非目標遮擋物遮擋后的人臉圖像，包括：

根據所述待組合人臉圖像上的關鍵點坐標、以及所述目標組合方式，確定所述待組合人臉圖像中的遮擋區域；

采用所述待組合非目標遮擋物的圖像填充所述遮擋區域，獲取所述非目標遮擋物遮擋后的人臉圖像。

7.根據權利要求6所述的方法，其特征在于，所述采用所述待組合非目標遮擋物的圖像填充所述遮擋區域，獲取所述非目標遮擋物遮擋后的人臉圖像，包括：

采用圖像處理方法，將所述待組合非目標遮擋物的圖像中對應位置的像素值拷貝到所述遮擋區域的對應位置，獲取所述非目標遮擋物遮擋后的人臉圖像。

8.一種遮擋物識別方法，其特征在于，包括：

獲取待識別圖像，所述待識別圖像包括含有遮擋物的人臉圖像；

采用遮擋物識別模型識別所述待識別圖像中的遮擋物是否為目標遮擋物，其中，所述遮擋物識別模型由人臉遮擋樣本訓練獲取，所述人臉遮擋樣本包括：不同非目標遮擋物遮擋后的人臉圖像以及所述非目標遮擋物的標簽，所述人臉遮擋樣本由人臉樣本集合、遮擋物負樣本集合以及預設組合方式組合獲取，所述人臉樣本集合包括：多個未遮擋人臉圖像；所述遮擋物負樣本集合包括：多種非目標遮擋物的圖像以及所述非目標遮擋物的標簽。