Python迅速從短視頻中獲取視頻幀的辦法詳細說明

89542767 發(fā)布于2022-12-21 17:18 / 812人閱讀

　　本文給大家介紹一類從視頻里獲取視頻幀的辦法，因為單核獲取視頻幀速率比較慢，因而接下來我們增強了線程同步的辦法，感興趣的朋友能夠出手試一試

　　Python迅速獲取視頻幀（線程同步）

　　現(xiàn)在詳細介紹一下一類從視頻里獲取視頻幀的辦法，因為單核獲取視頻幀速率比較慢，因而接下來我們增強了線程同步的辦法。

　　1、獲取視頻幀

　　獲取視頻幀關(guān)鍵用了Opencv控制模塊。

　　在其中：

　　camera=cv2.Videocapture()，函數(shù)公式主要通過啟用筆記本電腦內(nèi)置攝像頭載入視頻幀；

　　res,image=camera.read()函數(shù)公式通常是按幀載入短視頻，傳參“res”是布爾型，取得成功載入回到True，載入不成功回到False；

　　最終用cv2.imwrite()函數(shù)公式存放載入過的視頻幀。

　　視頻幀獲取方式可參考本文

　　import cv2
　　import os
　　def video_to_frames(video_path,outPutDirName):
　　times=0
　　#提取視頻的頻率，每1幀提取一個
　　frame_frequency=1
　　#如果文件目錄不存在則創(chuàng)建目錄
　　if not os.path.exists(outPutDirName):
　　os.makedirs(outPutDirName)
　　#讀取視頻幀
　　camera=cv2.VideoCapture(video_path)
　　while True:
　　times=times+1
　　res,image=camera.read()
　　if not res:
　　print('not res,not image')
　　break
　　#按照設(shè)置間隔存儲視頻幀
　　if times%frame_frequency==0:
　　cv2.imwrite(outPutDirName+''+str(times)+'.jpg',image)
　　print('圖片提取結(jié)束')
　　#釋放攝像頭設(shè)備
　　camera.release()

　　2、線程同步方式

　　線程同步的應(yīng)用關(guān)鍵應(yīng)用了threading庫。

　　在其中：

　　threading.Thread()函數(shù)公式主要是用于啟用線程同步，在其中主要參數(shù)“target”是上邊需要用到的函數(shù)公式，主要參數(shù)“args”是上邊函數(shù)的輸入數(shù)據(jù)。

　　在其中相關(guān)線程同步的詳細介紹一下，及其速率提高效果可參考本文

　　import threading
　　threading.Thread(target=video_to_frames,args=(video_path,outPutDirName)).start()

　　留意

　　1、extract_frame方方法的入?yún)⒎謩e是：鍵入視頻地址、導(dǎo)出視頻地址、短視頻fps、視頻尺寸寬、視頻尺寸高、短視頻必須抽走的起點幀、短視頻必須抽走的完畢幀。

　　3、整體代碼

　　import cv2
　　import os
　　import threading
　　def video_to_frames(video_path,outPutDirName):
　　times=0
　　#提取視頻的頻率，每1幀提取一個
　　frame_frequency=1
　　#如果文件目錄不存在則創(chuàng)建目錄
　　if not os.path.exists(outPutDirName):
　　os.makedirs(outPutDirName)
　　#讀取視頻幀
　　camera=cv2.VideoCapture(video_path)
　　while True:
　　times=times+1
　　res,image=camera.read()
　　if not res:
　　print('not res,not image')
　　break
　　if times%frame_frequency==0:
　　cv2.imwrite(outPutDirName+''+str(times)+'.jpg',image)
　　print('圖片提取結(jié)束')
　　camera.release()
　　if __name__=="__main__":
　　input_dir=r'D:datasetscow_dataset'#輸入的video文件夾位置
　　save_dir=r'E:relate_codedataset'#輸出圖片到當(dāng)前目錄video文件夾下
　　count=0#視頻數(shù)
　　for video_name in os.listdir(input_dir):
　　video_path=os.path.join(input_dir,video_name)
　　outPutDirName=os.path.join(save_dir,video_name[:-4])
　　threading.Thread(target=video_to_frames,args=(video_path,outPutDirName)).start()
　　count=count+1
　　print("%s th video has been finished!"%count)

　　補充

　　還可以利用Python實現(xiàn)抽取剔除視頻幀工具

　　代碼

　　下面是使用opencv對視頻中間幾幀抽取的方法。

　　主要的思路是在讀取frame的時候，順便把幀寫下來。

　　同時如果不是需要抽取剔除的幀，直接continue到下個循環(huán)。

　　樣例代碼如下，主要按照MP4格式進行處理。

　　#!/user/bin/env python
　　#coding=utf-8
　　"""
　　project:csdn-pro
　　author:劍客阿良_ALiang
　　file:test.py
　　ide:PyCharm
　　time:2022-06-30 17:55:48
　　"""
　　import cv2
　　#視頻抽幀
　　def extract_frame(video_path:str,result_path:str,fps,weight,height,start,end):
　　fourcc=cv2.VideoWriter_fourcc(*'mp4v')
　　videoWriter=cv2.VideoWriter(result_path,fourcc,fps,(weight,height))
　　vc=cv2.VideoCapture(video_path)
　　if vc.isOpened():
　　ret,frame=vc.read()
　　else:
　　ret=False
　　count=0#count the number of pictures
　　while ret:
　　ret,frame=vc.read()
　　if start&lt;=count&lt;=end:
　　count+=1
　　continue
　　else:
　　videoWriter.write(frame)
　　count+=1
　　print(count)
　　videoWriter.release()
　　vc.release()
　　if __name__=='__main__':
　　extract_frame('C:UsersxxxDesktop123.mp4','C:UsersxxxDesktop114.mp4',25,640,368,119,125)

　　綜上所述，這篇文章就給大家介紹到這里了，希望可以給大家?guī)韼椭?/p>