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

一、功能介绍

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

  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窗口。

二、代码

python 复制代码
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 += 1

frame_count = 0
current_frame = 0

while(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_interval
    frame_count += 1

print(f"Total {frame_count} frames extracted.")
cap.release()
cv2.destroyAllWindows()

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

相关推荐
来自宇宙的曹先生3 小时前
视频网站弹幕系统简易实现
spring boot·音视频
hjjdebug6 小时前
ffplay6 播放器关键技术点分析 1/2
c++·ffmpeg·音视频
_pengliang9 小时前
WebRTC 双向视频通话
音视频·webrtc
开开心心_Every10 小时前
全能视频处理工具介绍说明
开发语言·人工智能·django·pdf·flask·c#·音视频
沐尘而生15 小时前
【AI智能体】智能音视频-硬件设备基于 WebSocket 实现语音交互
大数据·人工智能·websocket·机器学习·ai作画·音视频·娱乐
nightunderblackcat18 小时前
进阶向:Python音频录制与分析系统详解,从原理到实践
开发语言·python·音视频
Antonio9151 天前
【音视频】HLS简介与服务器搭建
运维·服务器·音视频
开发者工具分享1 天前
文本音频违规识别工具排行榜(12选)
人工智能·音视频
开开心心_Every1 天前
便捷的Office批量转PDF工具
开发语言·人工智能·r语言·pdf·c#·音视频·symfony
karmueo462 天前
视频序列和射频信号多模态融合算法Fusion-Vital解读
算法·音视频·多模态