如何使用python實現批量視頻分幀、保存視頻幀

小編給大家分享一下如何使用python實現批量視頻分幀、保存視頻幀，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

利用python腳本實現視頻分幀，并將每一幀保存到本地。主要基于opencv包來實現，在運行代碼前確保opencv包已正確安裝。

代碼如下：

import os 
import cv2
 
videos_src_path = '/home/shao/violence_detection_code/Movies_Dataset/fights'
videos_save_path = '/home/shao/violence_detection_code/Movies_Dataset/fights_pic'
 
videos = os.listdir(videos_src_path)
videos.sort(key=lambda x:int(x[5:-4]))
 
i = 1
 
for each_video in videos:
 if not os.path.exists(videos_save_path + '/' + str(i)):
 os.mkdir(videos_save_path + '/' + str(i))
 each_video_save_full_path = os.path.join(videos_save_path,str(i))+'/'
 each_video_full_path = os.path.join(videos_src_path,each_video)
 cap = cv2.VideoCapture(each_video_full_path)
 frame_count = 1
 success = True
 
 while(success):
 success,frame = cap.read()
 if success==True:
 cv2.imwrite(each_video_save_full_path + "frame%d.jpg" % frame_count,
 frame)
 frame_count = frame_count + 1
 i = i + 1
 
 cap.release()

代碼注釋：

1.第1,2行完成導入幾個需要的包；

2.第4,5行分別是待分幀的原始視頻路徑、分幀后需要保存視頻幀的文件路徑；這里需要注意的是，我寫的是文件夾，也就是說在/home/shao/violence_detection_code/Movies_Dataset/fights目錄下有多個視頻XXX.avi,XXX.avi等形式；如果只需要處理單個視頻的話，可以稍微改一下即可。

3.第7行表示將所有視頻的文件名存到videos中，第8行是可選項，表示對fights文件夾下的視頻進行排序，x[5:-4]表示按照文件名第5個字符到倒數第4個字符之間的符號排序，因為我的視頻是newfi1.avi,newfi2.avi……的格式，我想實現的是按照數字1,2,3的順序提取視頻幀；

4.之后的代碼就是循環處理每個視頻，將每個視頻的視頻幀保存到文件夾中。

以上是“如何使用python實現批量視頻分幀、保存視頻幀”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！