[發明專利]行人圖像識別方法、裝置、電子設備及存儲介質

在審

• 申請號: 202111449400.7
• 申請日: 2021-11-30
• 公開（公告）號: CN114067277A
• 公開（公告）日: 2022-02-18
• 發明（設計）人: 鄧泳;張錦元;林曉銳;劉唱
• 申請（專利權）人: 中國工商銀行股份有限公司
• 主分類號: G06V20/52
• 分類號: G06V20/52;G06V40/16;G06V10/80;G06V10/774;G06V10/82;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京三友知識產權代理有限公司 11127
• 代理人: 孫乳筍;劉熔

1.一種行人圖像識別方法，其特征在于，所述方法包括：

獲取待檢測行人圖像，并對所述待檢測行人圖像進行預處理，該待檢測行人圖像包括：帶遮罩的人臉圖像；

將預處理后的圖像輸入至預先訓練的人臉檢測模型，以輸出人臉圖像，所述人臉檢測模型基于歷史行人圖像和經過數據增強處理的歷史行人圖像進行訓練，所述數據增加處理為對所述歷史行人圖像進行模板遮罩添加處理，該歷史行人的人臉圖像未帶遮罩；

將輸出的人臉圖像輸入至預先訓練的多屬性識別模型，以輸出與該人臉圖像相應的多個屬性，所述多屬性識別模型基于人臉圖像的多粒度特征識別人臉圖像的屬性。

2.根據權利要求1所述的方法，其特征在于，通過如下方式訓練所述人臉檢測模型：

獲取歷史行人圖像，并根據預定算法確定所述歷史行人圖像中的歷史人臉圖像中的關鍵點；

根據所述歷史人臉圖像中的關鍵點和預先構建的模板遮罩庫對所述歷史人臉圖像進行數據增強處理，以生成歷史模板遮罩人臉圖像；

根據所述歷史行人圖像和所述歷史模板遮罩人臉圖像訓練所述人臉檢測模型。

3.根據權利要求2所述的方法，其特征在于，根據所述歷史人臉圖像中的關鍵點和預先構建的模板遮罩庫對所述歷史人臉圖像進行數據增強處理，以生成歷史模板遮罩人臉圖像包括：

根據所述歷史人臉圖像中的關鍵點位置和關鍵點之間的距離確定該歷史人臉的傾斜角度；

根據所述歷史人臉圖像中的關鍵點和該歷史人臉的傾斜角度從所述模板遮罩庫選擇模板遮罩；

將選擇的模板遮罩添加到所述歷史人臉圖像上，以生成所述歷史模板遮罩人臉圖像。

4.根據權利要求2所述的方法，其特征在于，通過如下方式構建所述模板遮罩庫：

獲取初始遮罩圖像；

根據預先確定的人臉圖像中的關鍵點位置對所述初始遮罩圖像進行標定處理，以生成透明背景遮罩；

對所述透明背景遮罩進行幾何變換和色彩調整處理，以生成不同類別、色彩和角度的模板遮罩；

根據生成的不同類別、色彩和角度的模板遮罩構建模板遮罩庫。

5.根據權利要求2所述的方法，其特征在于，所述多屬性識別模型包括多粒度網絡，通過如下方式訓練所述多屬性識別模型：

獲取已標注屬性的歷史行人圖像和已標注人臉區域的歷史行人圖像；

基于圖神經網絡技術，并根據所述已標注屬性的歷史行人圖像、已標注人臉區域的歷史行人圖像和所述歷史模板遮罩人臉圖像訓練所述多屬性識別模型。

6.根據權利要求5所述的方法，其特征在于，將輸出的人臉圖像輸入至預先訓練的多屬性識別模型，以輸出與該人臉圖像相應的多個屬性包括：

將輸出的人臉圖像輸入至所述預先訓練的多屬性識別模型中的多粒度網絡，以提取該人臉圖像的多粒度特征；

基于圖神經網絡技術，對所述多粒度特征之間進行關聯建模操作，以輸出該人臉圖像相應的多個屬性。

7.一種行人圖像識別裝置，其特征在于，所述裝置包括：

待檢測圖像獲取單元，用于獲取待檢測行人圖像，并對所述待檢測行人圖像進行預處理，該待檢測行人圖像包括：帶遮罩的人臉圖像；

人臉圖像輸出單元，用于將預處理后的圖像輸入至預先訓練的人臉檢測模型，以輸出人臉圖像，所述人臉檢測模型基于歷史行人圖像和經過數據增強處理的歷史行人圖像進行訓練，所述數據增加處理為對所述歷史行人圖像進行模板遮罩添加處理，該歷史行人的人臉圖像未帶遮罩；

屬性識別單元，用于將輸出的人臉圖像輸入至預先訓練的多屬性識別模型，以輸出與該人臉圖像相應的多個屬性，所述多屬性識別模型基于人臉圖像的多粒度特征識別人臉圖像的屬性。