[發明專利]一種視頻的視角分類方法、裝置、設備及存儲介質

在審

• 申請號: 202110008444.X
• 申請日: 2021-01-05
• 公開（公告）號: CN112668533A
• 公開（公告）日: 2021-04-16
• 發明（設計）人: 彭聯貼;熊敏君;崔宵洋;顏家云;黎孟;張慧源;李晨;劉昕武;褚金鵬;劉邦繁;孫木蘭;劉雨聰
• 申請（專利權）人: 株洲中車時代電氣股份有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62;G06N3/08
• 代理公司: 北京集佳知識產權代理有限公司 11227
• 代理人: 侯珊

本發明公開了一種視頻的視角分類方法，在本方法中，對目標視頻的視角進行分類時，首先需要從目標視頻中抽取多幀圖像，并輸入預先訓練的單幀圖像分類模型，得到每幀圖像屬于不同視角的概率值，基于每幀圖像間的特異性特征及共性特征獲得目標視頻屬于各個視角的目標概率值，并通過將屬于同一視角的目標概率值與概率值閾值進行比較的方式，自動獲得目標視頻的視角分類結果；可見，本申請可實現對視頻視角的自動分類，減少了人工分類的工作，提高了分類速度；并且，本申請還考慮到了不同幀之間的共性特征及特異性特征，實現對視角的精準分類；本發明還公開了一種視頻視角分類裝置、設備及可讀存儲介質。

技術領域

本發明涉及圖像處理技術領域，更具體地說，涉及一種視頻的視角分類方法、裝置、設備及存儲介質。

背景技術

近年來，隨著視頻分析技術的發展，關于機車司機行為的分析識別正逐步由人工分析向智能分析轉變，其中的一種智能分析方式需要借助已有機車車載安全防護系統(6A系統)中的視頻監控設備。

目前，基于6A視頻的司機行為識別系統在前期的試用中有著不錯的效果，但都是以分析類似視角的視頻為前提，而機車上6A攝像頭安裝位置、角度的差異，會導致6A視頻會有不同的視角，以和諧電力機車為例，普遍存在右方和右后方這兩種視角畫面，以同一模型去分析不同視角的畫面，效果并不理想。因此，為了提升識別司機行為等智能分析系統的效果，如何對視頻視角進行分類是本領域技術人員需要解決的問題。

發明內容

本發明的目的在于提供一種視頻的視角分類方法、裝置、設備及存儲介質，以實現對視頻視角進行準確分類。

為實現上述目的，本發明提供一種視頻的視角分類方法，包括：

獲取目標視頻；

以預定時長為時間間隔從所述目標視頻中抽取第一預定數量幀的圖像；

將每幀圖像輸入預先訓練的單幀圖像分類模型，得到每幀圖像屬于不同視角的概率值；

基于圖像間的特異性特征及共性特征，利用所述目標視頻中每幀圖像屬于不同視角的概率值，計算所述目標視頻屬于每個視角的目標概率值；

將所述目標視頻屬于每個視角的目標概率值與對應視角的概率值閾值進行比較，根據比較結果確定目標視頻的視角分類結果。

其中，所述獲取目標視頻之前，還包括：

確定與每個視角對應的概率值閾值。

其中，所述確定與每個視角對應的概率值閾值，包括：

確定訓練視頻；其中，每個訓練視頻均具有已識別的視角；

以預定時長為時間間隔從每個訓練視頻中抽取第二預定數量幀的訓練圖像，并通過所述單幀圖像分類模型獲得每幀訓練圖像屬于不同視角的概率值；

對每個訓練視頻中各幀訓練圖像屬于對應視角的概率值進行更新；

利用所述更新后的概率值，計算與每個視角對應的每個訓練視頻的第一總概率值；所述第一總概率值為同一訓練視頻中各幀訓練圖像屬于對應視角的更新后的概率值之和；

根據所述第一總概率值以及所述第二預定數量幀確定與每個視角對應的概率值閾值。

其中，所述對每個訓練視頻中各幀訓練圖像屬于對應視角的概率值進行更新，包括：

獲取當前訓練視頻的每幀訓練圖像屬于所述當前訓練視頻的視角的訓練概率值，并判斷所述訓練概率值是否大于第一判定閾值；

若大于，則保留訓練圖像的概率值，否則，則將訓練圖像的概率值修改為零，以得到每個訓練視頻的每幀訓練圖像屬于對應視角的更新后的概率值。