[發(fā)明專利]視頻抽幀方法、裝置、電子設(shè)備、計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)

有效

• 申請?zhí)枺?/td>: 202110556431.6
• 申請日: 2021-05-21
• 公開（公告）號(hào): CN113286174B
• 公開（公告）日: 2022-11-08
• 發(fā)明（設(shè)計(jì)）人: 焦飛
• 申請（專利權(quán)）人: 浙江商湯科技開發(fā)有限公司
• 主分類號(hào): H04N21/234
• 分類號(hào): H04N21/234;H04N21/44
• 代理公司: 深圳市威世博知識(shí)產(chǎn)權(quán)代理事務(wù)所(普通合伙) 44280
• 代理人: 何倚雯

本發(fā)明提供一種視頻抽幀補(bǔ)償方法、裝置、電子設(shè)備以及計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，視頻抽幀方法包括：獲得多個(gè)視頻組；利用多個(gè)處理線程并行對所述多個(gè)視頻組進(jìn)行抽幀。以此提高視頻抽幀效率。

技術(shù)領(lǐng)域

本發(fā)明涉及計(jì)算機(jī)視覺技術(shù)領(lǐng)域，尤其是涉及一種視頻抽幀方法、裝置、電子設(shè)備、計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)。

背景技術(shù)

在計(jì)算機(jī)視覺領(lǐng)域中，訓(xùn)練模型時(shí)通常依賴大量的圖片作為輸入數(shù)據(jù)，而圖片則經(jīng)常是通過龐大的視頻文件按照一定策略進(jìn)行抽幀得到，例如按時(shí)間均勻抽幀，跳幀提取進(jìn)行視頻壓縮，按時(shí)間段抽幀，抽取關(guān)鍵幀等。傳統(tǒng)的方式是常用開源工具進(jìn)行視頻抽幀，這種方式存在一些弊端，例如CPU資源利用率不充分導(dǎo)致抽幀效率低，因此需要一種能夠提高抽幀效率的方法。

發(fā)明內(nèi)容

本發(fā)明提供一種視頻抽幀方法、裝置、電子設(shè)備以及計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，用以提高抽幀效率。

為解決上述技術(shù)問題，本發(fā)明提供的第一個(gè)技術(shù)方案為：提供一種視頻抽幀方法，包括：獲得多個(gè)視頻組；利用多個(gè)處理線程并行對所述多個(gè)視頻組進(jìn)行抽幀。多個(gè)處理線程并行對多個(gè)視頻組進(jìn)行抽幀，提高視頻抽幀效率。

其中，所述利用多個(gè)處理線程并行對所述多個(gè)視頻組進(jìn)行抽幀之前包括：獲取視頻處理服務(wù)器端的參數(shù)信息；根據(jù)所述參數(shù)信息得到所述多個(gè)處理線程；或者根據(jù)所述參數(shù)信息、所述多個(gè)視頻組的數(shù)量，得到所述多個(gè)處理線程。

多個(gè)處理線程并行對多個(gè)視頻組進(jìn)行抽幀，提高視頻抽幀效率。

所述獲取多個(gè)視頻組包括：獲取視頻處理服務(wù)器端的參數(shù)信息；根據(jù)所述參數(shù)信息對視頻進(jìn)行分組，以得到所述多個(gè)視頻組。減小視頻的規(guī)模，提高視頻抽幀效率。

所述參數(shù)信息包括所述視頻處理服務(wù)器的CPU總核數(shù)以及總內(nèi)存。

所述根據(jù)所述參數(shù)信息得到所述多個(gè)處理線程包括：根據(jù)所述視頻處理服務(wù)器的CPU總核數(shù)以及所述總內(nèi)存得到所述多個(gè)處理線程，每一所述處理線程利用預(yù)設(shè)數(shù)量的CPU核數(shù)處理一個(gè)視頻組。充分利用CPU，進(jìn)而提高視頻抽幀效率。

其中，所述利用多個(gè)處理線程并行對所述多個(gè)視頻組進(jìn)行抽幀包括：利用多個(gè)所述處理線程對所述多個(gè)視頻組進(jìn)行一一并行對應(yīng)處理。通過充分利用視頻處理服務(wù)器端的CPU，以提高視頻抽幀效率。

其中，每一所述處理線程包括第一子線程以及第二子線程，且每一所述處理線程包括寄存器；所述利用多個(gè)處理線程并行對所述多個(gè)視頻組進(jìn)行抽幀包括：利用每一所述第一子線程對每一所述視頻組進(jìn)行抽幀處理，以得到目標(biāo)幀，所述目標(biāo)幀存儲(chǔ)于所述寄存器中；從所述寄存器中獲取所述目標(biāo)幀，并對所述目標(biāo)幀進(jìn)行圖像處理，以形成目標(biāo)幀隊(duì)列，所述目標(biāo)幀隊(duì)列存儲(chǔ)于所述寄存器中；利用每一所述第二子線程從所述寄存器中讀取對應(yīng)的所述目標(biāo)幀隊(duì)列中的所述目標(biāo)幀，并存儲(chǔ)至存儲(chǔ)器中。在抽幀的過程中將目標(biāo)幀存儲(chǔ)在寄存器中，利用圖像處理器GPU對目標(biāo)幀進(jìn)行圖像處理，在處理完成后再寫入存儲(chǔ)器，無需將目標(biāo)幀寫入存儲(chǔ)器，再從存儲(chǔ)器中讀取目標(biāo)幀進(jìn)行圖像處理，降低了IO開銷，進(jìn)一步提高視頻抽幀效率，并提高圖像質(zhì)量。

其中，所述圖像處理包括圖像特征提取、圖像渲染中至少一種。

為解決上述技術(shù)問題，本發(fā)明提供的第二個(gè)技術(shù)方案為：提供一種視頻抽幀裝置，包括：獲取模塊，用于獲得多個(gè)視頻組；處理模塊，利用多個(gè)處理線程并行對所述多個(gè)視頻組進(jìn)行抽幀。

為解決上述技術(shù)問題，本發(fā)明提供的第三個(gè)技術(shù)方案為：提供一種電子設(shè)備，包括：存儲(chǔ)器及處理器，其中，所述存儲(chǔ)器存儲(chǔ)有程序指令，所述處理器從所述存儲(chǔ)器調(diào)取所述程序指令以執(zhí)行上述任一項(xiàng)所述的視頻抽幀方法。

為解決上述技術(shù)問題，本發(fā)明提供的第四個(gè)技術(shù)方案為：提供一種計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，存儲(chǔ)有程序文件，所述程序文件能夠被執(zhí)行以實(shí)現(xiàn)如上述任一項(xiàng)所述的視頻抽幀方法。