第一段代码：import os import cv2 def decode_video(video_path, save_dir, target_num=None): ''' video_path: 待解码的视频 save_dir: 抽帧图片的保存文件夹 target_num: 抽帧的数量, 为空则解码全部帧, 默认抽全部帧 ''' if not os.path.exists(save_dir): os.makedirs(save_dir) video = cv2.VideoCapture() if not video.open(video_path): print("can not open the video") exit(1) count = 0 index = 0 frames_num = video.get(7) # 如果target_num为空就全部抽帧,不为空就抽target_num帧 if target_num is None: step = 1 print('all frame num is {}, decode all'.format(int(frames_num))) else: step = int(frames_num/target_num) print('all frame num is {}, decode sample num is {}'.format(int(frames_num), int(target_num))) while True: _, frame = video.read() if frame is None: break if count % step == 0: save_path = "{}/{:>04d}.png".format(save_dir, index) cv2.imwrite(save_path, frame) index += 1 count += 1 if index == frames_num and target_num is None: # 如果全部抽，抽到所有帧的最后一帧就停止 break elif index == target_num and target_num is not None: # 如果采样抽，抽到target_num就停止 break else: pass video.release() if __name__ == '__main__': video_path = './test.mp4' save_dir_1 = './images_all' save_dir_2 = './images_sample' decode_video(video_path, save_dir_1) decode_video(video_path, save_dir_2, 20)

这是一个用于解码视频并抽取帧的代码。它使用了OpenCV库来处理视频文件。代码的主要功能是将视频文件解码并将抽取的帧保存为图像文件。

代码中的`decode_video`函数接受三个参数：`video_path`表示待解码的视频文件路径，`save_dir`表示抽帧图片的保存文件夹路径，`target_num`表示抽取的帧数量（可选参数，默认为None，即抽取全部帧）。

首先，代码会检查保存文件夹路径是否存在，如果不存在，则创建该文件夹。然后，通过`cv2.VideoCapture()`函数打开视频文件，并检查是否成功打开。如果无法打开视频文件，则输出错误信息并退出。

接下来，代码会初始化一些变量，包括计数器`count`、帧索引`index`以及视频的总帧数`frames_num`。根据`target_num`的值，代码会确定每隔多少帧抽取一帧，并输出相应的提示信息。

然后，代码会开始循环读取视频的每一帧。在每一次循环中，判断当前帧是否为空。如果为空，则说明已经读取完所有帧，循环结束。否则，根据计数器的值判断是否抽取当前帧，并将抽取的帧保存为图像文件。同时更新计数器和帧索引。

最后，根据抽取的帧数量和目标数量的关系，决定是否终止循环。当抽取全部帧时，如果帧索引等于总帧数，则停止循环。当采样抽取时，如果帧索引等于目标数量，则停止循环。

最后，释放视频对象的资源。

在代码的主程序部分，给定了一个测试视频文件路径`video_path`和两个保存文件夹路径`save_dir_1`和`save_dir_2`。分别调用了`decode_video`函数来进行全部帧抽取和采样抽取（抽取20帧）操作。

请注意，这只是一个示例代码，你可以根据自己的需求进行修改和扩展。