以下是一个使用Python编写的视频转图片程序，用户可以自定义从视频中提取的图片张数。我们将使用`opencv-python`库来处理视频和图像。 ```python import cv2 import os def video_to_frames(video_path, output_dir, num_frames): """ 将视频转换为指定数量的图片帧。 参数: - video_path: 视频文件路径 - output_dir: 输出图片保存的目录 - num_frames: 需要提取的图片帧数量 """ # 检查输出目录是否存在，不存在则创建 if not os.path.exists(output_dir): os.makedirs(output_dir) # 打开视频文件 cap = cv2.VideoCapture(video_path) total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) # 获取视频总帧数 fps = cap.get(cv2.CAP_PROP_FPS) # 获取视频帧率 print(f"视频总帧数: {total_frames}, 帧率: {fps} FPS") if num_frames > total_frames: print("所需帧数超过视频总帧数，将提取所有帧。") num_frames = total_frames # 计算需要提取的帧索引 frame_indices = [int(x * (total_frames / num_frames)) for x in range(num_frames)] count = 0 frame_num = 0 while True: ret, frame = cap.read() if not ret: break # 如果读取失败，跳出循环 if frame_num in frame_indices: # 保存当前帧为图片 image_path = os.path.join(output_dir, f"frame_{count:04d}.jpg") cv2.imwrite(image_path, frame) print(f"已保存: {image_path}") count += 1 frame_num += 1 cap.release() print(f"共提取了 {count} 张图片。") # 示例用法 if __name__ == "__main__": video_path = "example.mp4" # 输入视频路径 output_dir = "frames_output" # 输出图片保存目录 num_frames = 30 # 自定义提取的图片张数 video_to_frames(video_path, output_dir, num_frames) ``` ### 上述代码解释： 1. **导入库**：我们使用`cv2`（OpenCV）进行视频和图像处理，`os`用于文件路径操作。 2. **函数定义**：`video_to_frames`函数接受三个参数： - `video_path`: 视频文件路径。 - `output_dir`: 输出图片保存的目录。 - `num_frames`: 需要提取的图片帧数量。 3. **检查输出目录**：如果输出目录不存在，则自动创建。 4. **打开视频文件**：通过`cv2.VideoCapture`打开视频文件，并获取视频的总帧数和帧率。 5. **计算帧索引**：根据用户指定的帧数，均匀地选择需要提取的帧索引。 6. **逐帧读取**：遍历视频的所有帧，当当前帧的索引在选定的帧索引列表中时，保存该帧为图片。 7. **保存图片**：使用`cv2.imwrite`保存图片到指定目录。 ---