Python截取视频指定时段并生成超压缩GIF

pip install opencv-python pillow numpy

import cv2
import numpy as np
from PIL import Image
import os

def video_to_compressed_gif(
    video_path: str,
    save_gif_path: str,
    start_sec: float,
    end_sec: float,
    fps: int = 8,          # GIF帧率，越小体积越小
    resize_scale: float = 0.5,  # 画面缩放比例
    optimize: bool = True, # 开启Pillow GIF专属优化
    colors: int = 64       # 调色板色彩数量 2-256
):
    """
    视频指定时间段生成压缩GIF
    :param video_path: 原视频绝对/相对路径
    :param save_gif_path: 输出GIF保存路径及文件名
    :param start_sec: 截取起始时间（秒）
    :param end_sec: 截取结束时间（秒）
    :param fps: 输出GIF帧率
    :param resize_scale: 画面缩放系数（0-1）
    :param optimize: 开启GIF压缩优化
    :param colors: 自适应调色板色彩数量
    """
    # 打开视频文件
    cap = cv2.VideoCapture(video_path)
    if not cap.isOpened():
        raise FileNotFoundError(f"无法打开视频文件：{video_path}，请检查路径是否正确")

    # 获取视频原始参数
    video_fps = cap.get(cv2.CAP_PROP_FPS)
    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
    total_frames = cap.get(cv2.CAP_PROP_FRAME_COUNT)
    total_duration = total_frames / video_fps

    # 合法性参数校验
    if start_sec >= end_sec:
        cap.release()
        raise ValueError("错误：起始时间不能大于或等于结束时间")
    # 自动修正超出视频时长的参数
    if end_sec > total_duration:
        print(f"警告：结束时间超出视频总时长{total_duration:.2f}s，已自动截断至视频末尾")
        end_sec = total_duration

    # 计算帧抽取间隔，实现降帧压缩
    frame_interval = int(video_fps / fps)
    start_frame = int(start_sec * video_fps)
    cap.set(cv2.CAP_PROP_POS_FRAMES, start_frame)

    frame_list = []
    current_frame_idx = start_frame

    # 逐帧读取并处理画面
    while True:
        ret, frame = cap.read()
        if not ret:
            break
        current_time = current_frame_idx / video_fps
        if current_time > end_sec:
            break

        # 间隔抽帧，减少总帧数
        if (current_frame_idx - start_frame) % frame_interval == 0:
            # OpenCV默认BGR转RGB，适配PIL图像格式
            rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
            img = Image.fromarray(rgb_frame)
            # 分辨率缩放
            new_w = int(width * resize_scale)
            new_h = int(height * resize_scale)
            img = img.resize((new_w, new_h), Image.Resampling.LANCZOS)
            # 色彩量化压缩，减少色彩冗余
            img = img.convert("P", palette=Image.Palette.ADAPTIVE, colors=colors)
            frame_list.append(img)

        current_frame_idx += 1

    cap.release()
    # 校验是否截取到画面
    if len(frame_list) == 0:
        raise RuntimeError("未截取到任何画面，请检查视频路径或截取时间参数")

    # 保存极致压缩GIF
    duration_ms = int(1000 / fps)
    frame_list[0].save(
        save_gif_path,
        save_all=True,
        append_images=frame_list[1:],
        duration=duration_ms,
        loop=0,        # 0=无限循环，1=播放一次
        optimize=optimize,
        disposal=2     # 透明帧刷新优化，进一步缩小体积
    )
    print("=" * 50)
    print(f"✅ GIF生成成功！")
    print(f"📁 保存路径：{os.path.abspath(save_gif_path)}")
    print(f"⏱️ 截取时长：{end_sec - start_sec:.2f} 秒")
    print(f"🎞️ 总帧数：{len(frame_list)}")
    print("=" * 50)


if __name__ == "__main__":
    # ==================== 自定义配置区（仅修改这里即可）====================
    VIDEO_FILE = "test.mp4"       # 你的视频路径（相对/绝对路径均可）
    OUTPUT_GIF = "output.gif"     # 输出GIF文件名
    START_TIME = 2.0              # 截取起始时间（秒）
    END_TIME = 6.0                # 截取结束时间（秒）

    # 压缩参数（按需调整，数值越小体积越小）
    GIF_FPS = 6                    # GIF帧率（推荐3-10）
    SCALE = 0.4                    # 画面缩放比例（0.2-0.8）
    COLOR_COUNT = 48              # 色彩数量（16-128）
    # ====================================================================

    # 执行转换
    video_to_compressed_gif(
        video_path=VIDEO_FILE,
        save_gif_path=OUTPUT_GIF,
        start_sec=START_TIME,
        end_sec=END_TIME,
        fps=GIF_FPS,
        resize_scale=SCALE,
        colors=COLOR_COUNT
    )

import os

def ffmpeg_compress_gif(vid, out, start, dur, width=400, fps=6):
    """
    FFmpeg高阶压缩GIF
    :param vid: 视频路径
    :param out: 输出GIF路径
    :param start: 起始时间
    :param dur: 截取时长
    :param width: 输出宽度
    :param fps: 帧率
    """
    cmd = (
        f'ffmpeg -ss {start} -t {dur} -i "{vid}" '
        f'-filter_complex "fps={fps},scale={width}:-1,split[v1][v2];'
        f'[v1]palettegen=max_colors=48[p];[v2][p]paletteuse=dither=none" '
        f'-y "{out}"'
    )
    os.system(cmd)

# 使用示例
# ffmpeg_compress_gif("test.mp4", "ffmpeg_output.gif", start=2, dur=4)