Video Folder Name: Test
Video Type: avi
New Image Folder: Test_re
Image Type: jpg
import sys sys.path.append('.') import cv2 import os import glob import re from fastprogress import progress_bar def extract_numbers(file_name): numbers = re.findall(r'(\d+)', file_name) return tuple(map(int, numbers)) work='Test' video_list = glob.glob(f'./{work}/*.avi') video_list = sorted(video_list, key=extract_numbers) print(video_list) video_name_list = os.listdir(f'./{work}') video_name_list = sorted(video_name_list, key=extract_numbers) for i, video in progress_bar(enumerate(video_list), total=len(video_list)): video_name = video_name_list[i].split('.')[0] print(f'--------------{i, video_name}------------------') if not os.path.exists(f"./{work}_re/{video_name}"): os.makedirs(f"./{work}_re/{video_name}") vidcap = cv2.VideoCapture(video) success,image = vidcap.read() count = 0 while success: cv2.imwrite(f"./{work}_re/{video_name}/%04d.jpg" % count, image) # save frame as JPEG file success,image = vidcap.read() count += 1 print("finish! convert video to frame")