[發明專利]一種用于堡壘機錄屏審計的視頻分類方法和系統

在審

• 申請號: 202210330582.4
• 申請日: 2022-03-30
• 公開（公告）號: CN114722240A
• 公開（公告）日: 2022-07-08
• 發明（設計）人: 蔡宇輝;李肯立;王棵枝;楊圣洪;楊志邦;余思洋
• 申請（專利權）人: 湖南匡安網絡技術有限公司
• 主分類號: G06F16/75
• 分類號: G06F16/75;G06K9/62;G06N3/04;G06V10/774;G06V10/82
• 代理公司: 武漢臻誠專利代理事務所(普通合伙) 42233
• 代理人: 宋業斌

1.一種用于堡壘機錄屏審計的視頻分類方法，其特征在于，包括以下步驟：

(1)客戶端使用屏幕抓取工具對其屏幕進行實時錄制，對錄制的視頻進行處理，將處理得到的視頻進行編碼壓縮以得到視頻流，并將該視頻流發送到服務端。

(2)服務端依次將來自客戶端的視頻流解碼得到視頻，并將解碼后的視頻輸入訓練好的時間片段網絡TSN中，以得到該視頻對應的特征向量；

(3)服務端依次將步驟(2)得到的視頻流中所有視頻對應的所有特征向量輸入訓練好的長短時記憶網絡LSTM中，以得到視頻分類結果。

2.根據權利要求1所述的用于堡壘機錄屏審計的視頻分類方法，其特征在于，

步驟(1)中對錄制的視頻進行處理，包括對該視頻進行打水印和/或加時間戳處理；

對視頻進行編碼壓縮采用的是H264編碼方式。

本步驟中將視頻流發送到服務端是采用實時消息協議RTMP。

3.根據權利要求1或2所述的用于堡壘機錄屏審計的視頻分類方法，其特征在于，TSN網絡是通過以下步驟訓練得到的：

(2-1)獲取用于運維安全審計系統的錄屏數據集，將該錄屏數據集按照8:2的比例劃分為訓練集和測試集，并為訓練集中的每個樣本添加對應的真實分類標簽；

(2-2)對TSN網絡的參數進行初始化，權重參數的初始值是使用標準差為0.1的截斷式正態分布輸出的隨機值，偏置參數的初始值設為0，初始學習率lr＝0.01，采用階梯性的學習策略，權重gamma＝0.1，采樣的幀數是300幀，特征維度是2048個；采用的聚合函數是top-k池化(pooling)，其中k的取值是5；

(2-3)將步驟(2-1)獲取的訓練集輸入到步驟(2-2)初始化后的TSN網絡中，以通過稀疏采樣獲取訓練集中每個樣本對應的RGB視頻序列和光流場圖像特征，將每個樣本對應的RGB視頻序列和光流場圖像特征進行合并，以得到該樣本對應的特征向量，并獲取TSN網絡的評估精度；

(2-4)重復迭代執行上述步驟(2-3)，直到評估精度達到最佳為止，從而得到初步訓練好的TSN網絡。

(2-5)利用步驟(2-1)得到的測試集對步驟(2-4)初步訓練好的TSN網絡進行測試，以得到最終訓練好的TSN網絡。

4.根據權利要求1至3中任意一項所述的用于堡壘機錄屏審計的視頻分類方法，其特征在于，樣本的分類標簽共有七種，包括用戶管理、資產管理、工單管理、運維管理、協議管理、端口管理、指令管理。

5.根據權利要求3所述的用于堡壘機錄屏審計的視頻分類方法，其特征在于，步驟(2-3)中評估精度的表達式為：

其中x_i表示訓練集中的第i個樣本，其中i∈[1，N]，N表示訓練集中的樣本總數，y_i表示步驟(1)中為第i個樣本添加的實際分類標簽，f為預測類別函數，這里使用的是softmax函數。