[發明專利]一種人臉識別網絡的訓練方法、裝置、設備及存儲介質

在審

• 申請號: 202010592664.7
• 申請日: 2020-06-24
• 公開（公告）號: CN111783606A
• 公開（公告）日: 2020-10-16
• 發明（設計）人: 楊馥魁
• 申請（專利權）人: 北京百度網訊科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06N3/04;G06N3/08
• 代理公司: 北京品源專利代理有限公司 11332
• 代理人: 孟金喆

本申請公開了一種人臉識別網絡的訓練方法、裝置、設備及存儲介質，涉及人工智能中深度學習、云計算以及計算機視覺領域，具體涉及戴口罩人臉識別方面。具體實現方案為：將訓練數據對中的遮擋人臉圖像輸入教師網絡，無遮擋人臉圖像輸入學生網絡，并根據所述教師網絡和所述學生網絡對所述訓練數據對的預測結果，交替訓練所述教師網絡和所述學生網絡；其中，所述訓練數據對由相同身份的遮擋人臉圖像和非遮擋人臉圖像構成；從訓練后的教師網絡和學生網絡中，確定通用人臉識別網絡。以解決人臉識別結果準確性失衡的問題，提高人臉識別的準確性。

技術領域

本申請實施例涉及圖像處理技術領域，尤其涉及人工智能中深度學習、云計算以及計算機視覺領域，具體涉及戴口罩人臉識別方面。

背景技術

人臉識別技術是基于人的臉部特征信息進行身份識別的一種生物識別技術，已廣泛應用于生活中的各個領域。目前，可兼容識別無遮擋人臉圖像和遮擋人臉圖像的人臉識別網絡已成為當前人臉識別技術的研究方向。但是，現有的網絡訓練方法訓練的人臉識別網絡，在識別無遮擋人臉圖像和遮擋人臉圖像時，存在識別結果準確性失衡的情況，例如，只能準確識別遮擋人臉圖像或無遮擋人臉圖像中的一種。嚴重影響人臉識別網絡的識別精度，亟需改進。

發明內容

本公開提供了一種人臉識別網絡的訓練方法、裝置、設備及存儲介質。

根據本公開的一方面，提供了一種人臉識別網絡的訓練方法，該方法包括：

將訓練數據對中的遮擋人臉圖像輸入教師網絡，無遮擋人臉圖像輸入學生網絡，并根據所述教師網絡和所述學生網絡對所述訓練數據對的預測結果，交替訓練所述教師網絡和所述學生網絡；其中，所述訓練數據對由相同身份的遮擋人臉圖像和非遮擋人臉圖像構成；

從訓練后的教師網絡和學生網絡中，確定通用人臉識別網絡。

根據本公開的另一方面，提供了一種人臉識別網絡的訓練裝置，該裝置包括：

網絡訓練模塊，用于將訓練數據對中的遮擋人臉圖像輸入教師網絡，無遮擋人臉圖像輸入學生網絡，并根據所述教師網絡和所述學生網絡對所述訓練數據對的預測結果，交替訓練所述教師網絡和所述學生網絡；其中，所述訓練數據對由相同身份的遮擋人臉圖像和非遮擋人臉圖像構成；

網絡確定模塊，用于從訓練后的教師網絡和學生網絡中，確定通用人臉識別網絡。

根據本公開的另一方面，提供了一種電子設備，該電子設備包括：

至少一個處理器；以及

與所述至少一個處理器通信連接的存儲器；其中，

所述存儲器存儲有可被所述至少一個處理器執行的指令，所述指令被所述至少一個處理器執行，以使所述至少一個處理器能夠執行本申請任一實施例所述的人臉識別網絡的訓練方法。

根據本公開的另一方面，提供了一種存儲有計算機指令的非瞬時計算機可讀存儲介質，其中，所述計算機指令用于使所述計算機執行本申請任一實施例所述的人臉識別網絡的訓練方法。

根據本申請的技術解決了現有技術人臉識別結果準確性失衡的問題，提高了人臉識別的準確性。

應當理解，本部分所描述的內容并非旨在標識本公開的實施例的關鍵或重要特征，也不用于限制本公開的范圍。本公開的其它特征將通過以下的說明書而變得容易理解。

附圖說明

附圖用于更好地理解本方案，不構成對本申請的限定。其中：

圖1是根據本申請實施例提供的一種人臉識別網絡的訓練方法的流程圖；

圖2A是根據本申請實施例提供的另一種人臉識別網絡的訓練方法的流程圖；