【Python快速入门和实践019】Python常用脚本-查看视频信息并压缩视频

一、功能介绍

        这段代码用于查看视频帧率，帧数以及大小，并且是否需要按照设定参数压缩原始视频。主要定义了两个主要函数：analyze_video 和 compress_video，以及一个主程序块来调用这些函数。下面是对这段代码功能的详细分析：

analyze_video 函数

此函数用于分析给定视频的基本属性，并打印出这些信息。

功能描述

• 打开视频文件。
• 从视频中提取关键属性，包括：
  • 帧数 (frame_count)。
  • 帧率 (frame_rate)。
  • 分辨率 (width, height)。
  • 视频文件的大小 (video_size)，单位为兆字节（MB）。
• 打印视频信息。
• 返回提取的属性值。

compress_video 函数

此函数用于根据指定的参数压缩视频，并将压缩后的视频保存到一个新的文件中。

功能描述

• 打开视频文件。
• 获取原始视频的帧率和分辨率。
• 根据用户输入或默认值设置新的帧率和分辨率。
• 初始化视频写入器。
• 逐帧读取视频，按需调整大小，并写入新的视频文件。
• 释放资源。
• 计算并打印压缩后视频的大小。

主程序块

这部分代码负责：

• 调用 analyze_video 函数来显示视频的信息。
• 提供用户界面，询问用户是否需要压缩视频。
• 如果用户选择压缩视频，则收集压缩参数并通过 compress_video 函数执行压缩操作。

总结

这段代码实现了以下功能：

1. 视频分析：读取视频文件并提取关键属性。
2. 视频压缩：根据用户指定的新帧率和分辨率对视频进行压缩，并将结果保存到新文件中。
3. 用户交互：通过命令行界面与用户交互，允许用户决定是否进行视频压缩，并指定压缩参数。

二、代码

import cv2
import os
from tqdm import tqdm


def analyze_video(video_path):
    # 打开视频文件
    cap = cv2.VideoCapture(video_path)

    # 获取视频属性
    frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
    frame_rate = cap.get(cv2.CAP_PROP_FPS)
    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
    video_size = os.path.getsize(video_path) / (1024 * 1024)  # 文件大小，单位MB

    cap.release()

    print(f"视频信息：\n"
          f"帧数: {frame_count}\n"
          f"帧率: {frame_rate} FPS\n"
          f"分辨率: {width} x {height}\n"
          f"大小: {video_size:.2f} MB")

    return frame_count, frame_rate, (width, height), video_size


def compress_video(input_video, output_video, frame_rate=None, resolution=None, codec='XVID'):
    # 打开视频文件
    cap = cv2.VideoCapture(input_video)

    # 获取原始视频信息
    original_frame_rate = cap.get(cv2.CAP_PROP_FPS)
    original_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
    original_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

    # 如果没有指定帧率或分辨率，则使用原始视频参数
    if frame_rate is None:
        frame_rate = original_frame_rate
    if resolution is None:
        resolution = (original_width, original_height)

    width, height = resolution

    # 初始化视频写入器
    fourcc = cv2.VideoWriter_fourcc(*codec)
    out = cv2.VideoWriter(output_video, fourcc, frame_rate, (width, height))

    # 压缩视频并保存
    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
    for _ in tqdm(range(total_frames), desc="Compressing video"):
        ret, frame = cap.read()
        if not ret:
            break

        # 调整大小
        resized_frame = cv2.resize(frame, (width, height))
        out.write(resized_frame)

    cap.release()
    out.release()

    output_size = os.path.getsize(output_video) / (1024 * 1024)  # 输出视频文件大小，单位MB
    print(f"压缩后的视频已保存到 {output_video}，大小: {output_size:.2f} MB")


if __name__ == "__main__":
    # 分析视频
    video_path = "demo.mp4"
    frame_count, frame_rate, resolution, video_size = analyze_video(video_path)

    # 询问用户是否需要压缩视频
    compress = input("是否要压缩视频？(y/n): ").strip().lower()

    if compress == 'y':
        # 设置压缩参数
        new_frame_rate = float(input(f"设置新帧率 (当前: {frame_rate} FPS, 按Enter跳过): ") or frame_rate)
        width = int(input(f"设置新宽度 (当前: {resolution[0]}, 按Enter跳过): ") or resolution[0])
        height = int(input(f"设置新高度 (当前: {resolution[1]}, 按Enter跳过): ") or resolution[1])

        output_video = "compressed_video.mp4"
        compress_video(video_path, output_video, frame_rate=new_frame_rate, resolution=(width, height))