[發明專利]人體圖像的分類方法和裝置、電子設備、存儲介質、程序

有效

• 申請號: 201711399693.6
• 申請日: 2017-12-21
• 公開（公告）號: CN108229353B
• 公開（公告）日: 2020-09-22
• 發明（設計）人: 李玉潔;曠章暉;張偉
• 申請（專利權）人: 深圳市商湯科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62;G06N3/02
• 代理公司: 北京思源智匯知識產權代理有限公司 11657
• 代理人: 毛麗琴

1.一種人體圖像的分類方法，其特征在于，包括：

分別對待處理圖像進行特征提取和掩碼提取操作，獲得對應所述待處理圖像的圖像特征和對應所述待處理圖像中人體區域的區域掩碼；所述待處理圖像中包括至少一個人體區域，所述區域掩碼用于突出所述待處理圖像中的人體區域；其中，所述區域掩碼的生成依賴于到人體關鍵點的距離，所述區域掩碼在所述人體關鍵點附近取值大，遠離關鍵點取值小；所述人體關鍵點用于標識待處理圖像中人體區域的位置；

融合所述圖像特征和所述區域掩碼，得到所述人體區域的區域增強特征；

基于分類網絡，根據所述區域增強特征獲得所述待處理圖像的分類結果。

2.根據權利要求1所述的方法，其特征在于，所述分別對待處理圖像進行特征提取和掩碼提取操作，獲得對應所述待處理圖像的圖像特征和對應所述待處理圖像中人體區域的區域掩碼，包括：

利用第一神經網絡對待處理圖像進行特征提取操作，獲得對應所述待處理圖像的圖像特征；

利用第二神經網絡對待處理圖像進行掩碼提取操作，獲得對應所述待處理圖像中人體區域的區域掩碼；所述圖像特征與所述區域掩碼在長和寬的維度上大小相同。

3.根據權利要求2所述的方法，其特征在于，所述利用第二神經網絡對待處理圖像進行掩碼提取操作，獲得對應所述待處理圖像中人體區域的區域掩碼，包括：

利用第二神經網絡對所述待處理圖像執行人體關鍵點檢測操作，獲得具有人體區域輪廓的關鍵點特征圖；

通過計算所述關鍵點特征圖中所有像素點與所有所述人體關鍵點的距離獲得所述區域掩碼。

4.根據權利要求3所述的方法，其特征在于，通過計算所述關鍵點特征圖中所有像素點與所有所述人體關鍵點的距離獲得所述區域掩碼，包括：

通過計算所述關鍵點特征圖中的像素點與所有所述人體關鍵點的距離，獲得所述像素點的目標距離值；

利用正態分布函數，基于所述像素點的目標距離值獲得所述像素點對應的掩碼值；

基于所述關鍵點特征圖中所有所述像素點對應的掩碼值構成所述區域掩碼。

5.根據權利要求4所述的方法，其特征在于，所述通過計算所述關鍵點特征圖中的像素點與所有所述人體關鍵點的距離，獲得所述像素點的目標距離值，包括：

計算所述關鍵點特征圖中的像素點與所述人體關鍵點的歐式距離，獲得多個距離值；

將所述距離值中的最小距離值作為所述像素點的目標距離值。

6.根據權利要求4所述的方法，其特征在于，所述利用正態分布函數，基于所述像素點的目標距離值獲得所述像素點對應的掩碼值，包括：

將所述像素點的目標距離值輸入正態分布函數，獲得對應所述像素點的掩碼中間值；

基于所述掩碼中間值計算得到對應所述像素點的掩碼值。

7.根據權利要求6所述的方法，其特征在于，所述基于掩碼中間值計算得到對應所述像素點的掩碼值，包括：

將所述得到的掩碼中間值加設定值得到對應的掩碼值；與所述人體關鍵點的距離小于預設值的像素點對應的掩碼值大于預設閾值，與所述人體關鍵點的距離大于預設值的像素點對應的掩碼值小于預設閾值。

8.根據權利要求1-7任一所述的方法，其特征在于，融合所述圖像特征和區域掩碼，得到所述人體區域的區域增強特征，包括：

將所述圖像特征中的元素與所述區域掩碼中對應的元素執行點乘運算，得到所述人體區域的圖像特征增強后的區域增強特征；所述圖像特征與所述區域掩碼在長度維度和寬度維度上的大小相對應，所述元素包括所述圖像特征對應的特征圖中的像素點或所述圖像特征對應的特征向量中的向量值。