python opencv视频帧提取

视频帧提取

import tensorflow as tf
import cv2
import os
videos_src_path=‘D:/300VW_Dataset_2015_12_14/300VW_Dataset_2015_12_14/001’
videos_save_path=‘D:/300VW_Dataset_2015_12_14/300VW_Dataset_2015_12_14/001/vid’

videos=os.listdir(videos_src_path)
videos=filter(lambda x: x.endswith(‘avi’),videos)
for each_video in videos:
print(each_video)
each_video_name,_=each_video.split(’.’)
os.mkdir(videos_save_path+’/’+each_video_name)
each_video_save_full_path=os.path.join(videos_save_path,each_video_name)+’/’
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()
print(‘read a new frame:’,success)
params=[]
params.append(cv2.IMWRITE_JPEG_QUALITY)
params.append(100)
cv2.imwrite(each_video_save_full_path+"%06d.jpg"%frame_count,frame,params)
frame_count=frame_count+1
cap.release()