本篇博客介紹利用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.之后的代碼就是循環處理每個視頻,將每個視頻的視頻幀保存到文件夾中。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索:z360901061

微信掃一掃加我為好友
QQ號聯系: 360901061
您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對您有幫助就好】元
