1 import cv2 2 import os 3 4 video_path = './video' 5 out_frame_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'frame_output') 6 if not os.path.exists(out_frame_path): 7 os.makedirs(out_frame_path) 8 print('out_frame_path', out_frame_path) 9 files = [] 10 list = os.listdir(video_path) 11 print('list', list) 12 for i in range(len(list)): 13 item = os.path.join(video_path, list[i]) 14 files.append(item) 15 print('files',files) 16 for k,file in enumerate(files): 17 frame_dir = os.path.join(out_frame_path, '%d'%(k+1)) 18 if not os.path.exists(frame_dir): 19 os.makedirs(frame_dir) 20 cap = cv2.VideoCapture(file) 21 j = 0 22 print('start prossing NO.%d video' % (k + 1)) 23 while True: 24 ret, frame = cap.read() 25 if ret: 26 cv2.imwrite(os.path.join(frame_dir, '%d.jpg'%j), frame) 27 j += 1 28 else: 29 cap.release() 30 break 31 print('prossed NO.%d video'%(k+1))