Python将视频转换为图片介绍

前言

代码让人头秃，可是我喜欢上了代码的逻辑感和代码解决问题的力量。
（我要对这个专题的图片进行修改，缩小看和我导头像有点像QAQ）

代码展示

此段代码能够实现多个文件夹中多个视频片段的转换，代码的具体含义可以参考注释。
具体操作为将此段代码复制粘贴到创建好的main.py文件，配置好第63行的源文件路径，然后运行。

# 将四个文件夹中的多个视频文件转换为图片import cv2import osnumber_in_dir = []  # 可以去掉def video2image(video_path, image_save_dir):# 1. 将图片的存储路径加上源文件的文件夹名，如\'Coffee_room_01\'等。if os.path.exists(video_path):  # 判断源路径是否正确print(video_path + \'\\t ok\')if video_path.split(\'\\\\\')[-2] != \'Videos\':  # 这个根据我文件路径自行添加的image_save_dir += video_path.split(\'\\\\\')[-2] + \'\\\\\'else:image_save_dir += video_path.split(\'\\\\\')[-3] + \'\\\\\'number_in_dir.append(len(os.listdir(video_path)))  # 可以去掉else:print(video_path + \' \\033[0;37;41merror\\033[0m\')return# 2. 依次读取源文件里的文件，如果后缀名是‘avi\'或 \'MP3\'，则创建一个关于文件名的子文件夹files_in_video_path_list = os.listdir(video_path)print(len(files_in_video_path_list))  # 获取当前文件夹下文件个数，用来判断是否有其他类型的文件file_count = 0  # 用于统计个数，验证是否全为视频文件，会与len(files_in_video_path_list)进行比较for file_in_video_path in files_in_video_path_list:file_name = os.path.basename(file_in_video_path)if file_name.split(\'.\')[-1] == \'avi\' or file_name.split(\'.\')[-1] == \'mp3\':file_count += 1  # 视频文件数+1every_video_save_dir = file_name.split(\'.\')[0] + \'\\\\\'if not os.path.exists(image_save_dir + every_video_save_dir):  # 创建属于相应文件夹的存储路径os.makedirs(image_save_dir + every_video_save_dir)else:print(\' \\033[0;37;41merror\\033[0m\')break# 3. 打印正在处理文件的序号和他的文件名，并开始转换print(str(file_count) + \':\' + file_name)cap = cv2.VideoCapture(video_path + file_name)flag = cap.isOpened()if not flag:print(\"open\" + video_path + file_name + \"error!\")frame_count = 0  # 给每一帧标号while True:frame_count += 1flag, frame = cap.read()if not flag:  # 如果已经读取到最后一帧则退出breakif os.path.exists(image_save_dir + every_video_save_dir + str(frame_count) + \'.jpg\'):  # 在源视频不变的情况下，如果已经创建，则跳过breakcv2.imwrite(image_save_dir + every_video_save_dir + str(frame_count) + \'.jpg\', frame)cap.release()print(file_name + \' save to \' + image_save_dir + every_video_save_dir + \'finished \')  # 表示一个视频片段已经转换完成if file_count != len(files_in_video_path_list):print(\' \\033[0;37;41Maybe loss something\\033[0m\')if __name__ == \'__main__\':# 需要转换的文件路径列表(自定义修改）video_path_list = [\'E:\\\\Datasets_FallDetection\\\\2 LE2I\\Coffee_room_01\\\\Coffee_room_01\\\\Videos\\\\\',\'E:\\\\Datasets_FallDetection\\\\2 LE2I\\Home_01\\\\Home_01\\Videos\\\\\',\'E:\\\\Datasets_FallDetection\\\\2 LE2I\\Lecture_room\\\\Lecture room\\\\\',\'E:\\\\Datasets_FallDetection\\\\2 LE2I\\\\Office\\\\Office\\\\\']# 预期存储在的主文件夹，即\'result\'文件夹image_save_dir = \'.\\\\result\\\\\'# 进行转换for video_path in video_path_list:video2image(video_path, image_save_dir)