当前位置: 首页> 健康> 养生 > 【Python快速入门和实践016】Python常用脚本-对视频抽取指定帧数并保存

【Python快速入门和实践016】Python常用脚本-对视频抽取指定帧数并保存

时间:2025/7/11 8:11:52来源:https://blog.csdn.net/2403_83044722/article/details/141230285 浏览次数:2次

一、功能介绍

        这段代码的功能是从一个视频文件中抽取指定数量的帧,并将这些帧保存为图像文件。步骤如下:

  1. 设置路径和参数

    • video_path:视频文件的路径。
    • image_folder:保存抽取图像的目录。
    • num_frames_to_extract:需要抽取的帧数。
  2. 创建输出目录

    • 如果指定的输出目录不存在,则创建该目录。
  3. 打开视频文件

    • 使用cv2.VideoCapture打开视频文件。
  4. 获取视频信息

    • 获取视频的帧率(fps)和总帧数(total_frames)。
  5. 计算抽取间隔

    • 计算抽取帧的间隔,以确保尽可能均匀地从视频中抽取帧。
    • 如果总帧数不能被需要抽取的帧数整除,则增加1以确保最后一帧也能被抽取。
  6. 循环抽取帧

    • 循环读取视频帧,直到抽取的帧数达到指定数量或读取完视频。
    • 使用cap.set(cv2.CAP_PROP_POS_FRAMES, current_frame)跳转到指定帧。
    • 读取帧,并保存为图像文件。
  7. 保存图像

    • 每次读取到有效帧时,保存为JPEG格式的图像文件。
    • 文件名格式为normal-XXX.jpg,其中XXX是三位数的帧序号,不足三位数时前面补零。
  8. 释放资源

    • 关闭视频文件。
    • 清理OpenCV窗口。

二、代码

import cv2
import os# 视频文件路径
video_path = r'demo.mp4'# 输出图像目录
image_folder = r'E:\pythonProject\pythonProject\imgs_chouzhen'
if not os.path.exists(image_folder):os.makedirs(image_folder)# 指定要抽取的帧数
num_frames_to_extract = 100# 打开视频文件
cap = cv2.VideoCapture(video_path)# 获取视频总帧数
fps = cap.get(cv2.CAP_PROP_FPS)  # 帧率
total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))  # 总帧数# 计算抽取间隔(尽量均匀)
frame_interval = total_frames // num_frames_to_extract
if total_frames % num_frames_to_extract != 0:frame_interval += 1frame_count = 0
current_frame = 0while(current_frame < total_frames and frame_count < num_frames_to_extract):cap.set(cv2.CAP_PROP_POS_FRAMES, current_frame)ret, frame = cap.read()if not ret:break# 保存图像img_name = f"normal-{str(frame_count).zfill(3)}.jpg"img_path = os.path.join(image_folder, img_name)cv2.imwrite(img_path, frame)current_frame += frame_intervalframe_count += 1print(f"Total {frame_count} frames extracted.")
cap.release()
cv2.destroyAllWindows()

改成你的视频路径和保存路径,执行结果:

关键字:【Python快速入门和实践016】Python常用脚本-对视频抽取指定帧数并保存

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

责任编辑: