python使用OpenCV 库将视频拆解为帧并保存为图片

python使用OpenCV 库将视频拆解为帧并保存为图片

import cv2
import osdef video_to_frames(video_path, output_folder, frame_prefix='frame_', interval=1, target_size=None, grayscale=False):"""将视频拆分为帧并保存为图片参数:video_path (str): 视频文件路径output_folder (str): 输出图片的文件夹路径frame_prefix (str): 保存帧图片的前缀（默认: frame_）interval (int): 抽帧间隔（默认: 1，即所有帧）target_size (tuple): 目标尺寸 (宽, 高)，None表示保持原尺寸（默认）grayscale (bool): 是否转换为灰度图（默认: False）"""# 创建输出文件夹os.makedirs(output_folder, exist_ok=True)# 读取视频文件cap = cv2.VideoCapture(video_path)if not cap.isOpened():print("Error: 无法打开视频文件")returnframe_count = 0saved_count = 0while True:ret, frame = cap.read()if not ret:break  # 视频读取结束# 按间隔处理帧if frame_count % interval == 0:processed_frame = frame.copy()# 调整尺寸if target_size is not None:processed_frame = cv2.resize(processed_frame, target_size)# 转换为灰度图if grayscale:processed_frame = cv2.cvtColor(processed_frame, cv2.COLOR_BGR2GRAY)# 生成文件名filename = f"{frame_prefix}{saved_count:04d}.jpg"file_path = os.path.join(output_folder, filename)# 保存图片cv2.imwrite(file_path, processed_frame)saved_count += 1frame_count += 1# 释放资源cap.release()print(f"处理完成，共保存 {saved_count} 帧图片")print(f"视频总帧数: {frame_count}")# 使用示例
if __name__ == "__main__":video_path = r"E:\photo\video\1.mp4"  # 替换为你的视频路径output_folder = r"E:\photo\video\1"  # 输出文件夹# 示例1：提取所有帧（原始尺寸和颜色）video_to_frames(video_path, output_folder)# 示例2：调整尺寸为 640x480，保存为灰度图，每隔10帧保存一次# video_to_frames(#     video_path,#     output_folder,#     interval=10,#     target_size=(640, 480),#     grayscale=True# )# 示例3：仅调整尺寸为 1280x720# video_to_frames(#     video_path,#     output_folder,#     target_size=(1280, 720)# )

• target_size: 接收元组格式 (width, height)

• grayscale: 布尔值控制是否转为灰度图

• 先执行尺寸调整（保留更多原始信息）后执行灰度转换（减少最后一步的计算量）