[發明專利]一種行為檢測方法、裝置及電子設備

在審

• 申請號: 202010965429.X
• 申請日: 2020-09-14
• 公開（公告）號: CN112052815A
• 公開（公告）日: 2020-12-08
• 發明（設計）人: 史曉蒙;韓晴;張星;宋征;李高楊;魏健康;張偉
• 申請（專利權）人: 北京易華錄信息技術股份有限公司;中國華錄集團有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62;G06T7/187
• 代理公司: 北京三聚陽光知識產權代理有限公司 11250
• 代理人: 羅嘯

1.一種行為檢測方法，其特征在于，包括：

獲取目標監控區域的視頻數據；

利用預先訓練好的人臉識別模型對所述視頻數據中的圖像數據進行人臉檢測；

對檢測到人臉數據對應的框選區域進行擴展處理，使得擴展后的框選區域大于人臉所在的整個頭部區域；

利用預先訓練好的行為檢測模型，對所述擴展后的框選區域進行行為檢測，得到檢測結果，所述行為檢測模型根據包括整個頭部區域的違法行為人臉圖像數據訓練得到。

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

獲取多個實際場景下的人臉圖像數據，構建第一人臉圖像數據集，每一個場景下的人臉圖像數據包含不同時間段的圖像；

對所述第一人臉圖像數據集中的圖像進行數據增強處理，得到擴大數據量的第一人臉圖像數據集；

利用擴大數據量的第一人臉圖像數據集對第一目標網絡模型進行模型訓練，得到所述人臉識別模型。

3.根據權利要求2所述的方法，其特征在于，所述第一目標網絡模型包括RetinaFace人臉檢測模型。

4.根據權利要求1所述的方法，其特征在于，所述行為檢測模型通過下述方式訓練得到：

獲取多個實際場景下的違法行為人臉圖像數據和非違法行為人臉圖像數據，構建第二人臉圖像數據集，每一個場景下的人臉圖像數據包含不同時間段的圖像；

將所述第二人臉圖像數據集中的人臉圖像數據對應的框選區域進行擴展處理，使得擴展后的框選區域大于人臉所在的整個頭部區域；

對擴展后的所述第二人臉圖像數據集中的圖像進行數據增強處理，得到擴大數據量的第二人臉圖像數據集；

利用所述擴大數據量的第二人臉圖像數據集對第二目標網絡模型進行模型訓練，直至目標損失函數的損失值滿足預設條件，得到所述行為檢測模型。

5.根據權利要求4所述的方法，其特征在于，所述第二目標網絡模型包括：OSNet網絡模型，其中所述OSNet網絡模型的特征層與全連接層之間增加注意力機制；所述目標損失函數包括：NLLLOSS損失函數。

6.根據權利要求4所述的方法，其特征在于，所述利用所述擴大數據量的第二人臉圖像數據集對第二目標網絡模型進行模型訓練，包括：

將所述擴大數據量的第二人臉圖像數據集按照目標比例進行劃分，得到訓練集、驗證集以及測試集；

利用所述訓練集對所述第二目標網絡模型進行模型訓練。

7.根據權利要求1所述的方法，其特征在于，所述利用預先訓練好的人臉識別模型對所述視頻數據中的圖像數據進行人臉檢測之前，所述方法還包括：

對獲取到目標監控區域的視頻數據按照目標間隔時間進行抽幀處理，得到需要進行人臉檢測的圖像數據。

8.一種行為檢測裝置，其特征在于，包括：

獲取模塊，用于獲取目標監控區域的視頻數據；

第一檢測模塊，用于利用預先訓練好的人臉識別模型對所述視頻數據中的圖像數據進行人臉檢測；

擴展模塊，用于對檢測到人臉數據對應的框選區域進行擴展處理，使得擴展后的框選區域大于人臉所在的整個頭部區域；

第二檢測模塊，用于利用預先訓練好的行為檢測模型，對所述擴展后的框選區域進行行為檢測，得到檢測結果，所述行為檢測模型根據包括整個頭部區域的違法行為人臉圖像數據訓練得到。

9.一種電子設備，其特征在于，包括：至少一個處理器；以及與所述至少一個處理器通信連接的存儲器；其中，所述存儲器存儲有可被所述至少一個處理器執行的指令，所述指令被所述至少一個處理器執行，以使所述至少一個處理器執行如權利要求1-7任一所述的行為檢測方法的步驟。

10.一種計算機可讀存儲介質，其上存儲有計算機程序，其特征在于，所述計算機程序被處理器執行時實現如權利要求1-7中任一項所述的行為檢測方法的步驟。