如何用Python的imageio获取视频最后一帧及均匀分布的5帧？

一、准确获取视频总帧数的方法 #

不要用「时长×帧率」计算，因为视频的帧率可能是平均帧率（尤其是可变帧率视频），时长也可能是近似值，两者相乘的结果往往和实际帧数有偏差。推荐两种可靠方案：

1. 用FFprobe（FFmpeg工具集） #

直接执行命令获取精确的视频帧数量：

ffprobe -v error -select_streams v:0 -count_packets -show_entries stream=nb_read_packets -of csv=p=0 input.mp4

该命令会直接输出视频的实际总帧数，避免了计算误差。

2. 用OpenCV（Python） #

通过CAP_PROP_FRAME_COUNT属性获取，虽然少数编码格式下可能有微小误差，但大部分场景足够可靠：

import cv2
cap = cv2.VideoCapture("input.mp4")
total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
cap.release()

二、均匀取帧的实现逻辑 #

要提取5帧且包含首尾，核心是计算帧间隔：

• 设总帧数为N，则5帧之间有4个间隔，间隔值为(N-1) // (5-1)（用整数除法避免浮点误差）
• 取帧的索引为：0, (N-1)//4, 2*(N-1)//4, 3*(N-1)//4, N-1

这样既能保证首尾帧被选中，又能让所有帧均匀分布在整个视频时间线上。

三、高效实现方案 #

1. FFmpeg命令行（推荐，无需遍历全帧） #

直接通过滤镜定位目标帧，效率极高，无需读取所有帧：

ffmpeg -i input.mp4 -vf "select='eq(n,0)+eq(n,floor((nb_frames-1)/4))+eq(n,floor(2*(nb_frames-1)/4))+eq(n,floor(3*(nb_frames-1)/4))+eq(n,nb_frames-1)'" -vsync vfr frame_%d.jpg

• nb_frames是FFmpeg内置的视频总帧数变量，会自动获取
• -vsync vfr确保输出帧的时间戳不强制同步，避免重复或丢失帧
• 输出文件会按顺序命名为frame_0.jpg到frame_4.jpg

2. Python + OpenCV #

如果需要在代码中处理帧数据，可以用以下实现：

import cv2

def extract_uniform_frames(video_path, num_frames=5):
    cap = cv2.VideoCapture(video_path)
    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
    # 计算间隔，确保首尾帧被包含
    interval = (total_frames - 1) // (num_frames - 1)
    
    extracted_frames = []
    for idx in range(num_frames):
        target_frame = idx * interval
        # 定位到目标帧
        cap.set(cv2.CAP_PROP_POS_FRAMES, target_frame)
        ret, frame = cap.read()
        if ret:
            extracted_frames.append(frame)
            # 可选：保存帧到本地
            # cv2.imwrite(f"extracted_frame_{idx}.jpg", frame)
        else:
            print(f"无法读取第 {target_frame} 帧")
    
    cap.release()
    return extracted_frames

# 调用示例
frames = extract_uniform_frames("input.mp4")

注：如果遇到cap.set定位不准的情况（比如某些编码的视频），可以先检查视频的关键帧分布，或改用FFmpeg方案。

内容的提问来源于stack exchange，提问作者JasonTS