python视频抽帧

### 使用Python从视频中提取帧

对于计算机视觉项目而言，能够从视频文件中高效地抽取图像帧是一项重要技能。通过`opencv-python`库可以实现这一目标[^1]。

#### 安装所需库
为了执行此任务，需先安装OpenCV库。可以通过pip命令完成安装：

pip install opencv-python

#### 方法一：基于时间间隔抽取帧
下面这段代码展示了一个简单的例子，它会按照设定的时间间隔（例如每一秒）保存一张图片到本地磁盘上。

import cv2

def extract_frames_by_time(video_path, output_prefix='frame', fps=1):
    cap = cv2.VideoCapture(video_path)

    if not cap.isOpened():
        print("Error opening video file")

    frame_rate = cap.get(cv2.CAP_PROP_FPS)  # 获取原视频FPS
    interval = int(frame_rate / fps)       # 计算每隔多少帧采集一次
    
    count = 0
    success = True
    while success:
        success, image = cap.read()
        
        if (count % interval == 0):         # 当前帧号能被interval整除则保存该帧
            filename = f"{output_prefix}_{int(count/interval)}.jpg"
            cv2.imwrite(filename, image)
            
        count += 1
        
    cap.release()

extract_frames_by_time('example.mp4')

#### 方法二：按固定数量的帧间距离抽取
另一种方式是指定固定的帧间距来进行采样，在这种情况下不需要考虑实际播放速度的影响[^2]。

import cv2

def extract_frames_by_interval(video_path, output_prefix='frame', skip_frames=10):
    cap = cv2.VideoCapture(video_path)

    if not cap.isOpened():
        print("Error opening video file")

    index = 0
    counter = 0
    while(cap.isOpened()):
        ret, frame = cap.read()
        if ret == False:
            break
            
        if index % skip_frames == 0:          # 如果当前索引可被skip_frames整除，则写入文件
            name = './data/' + str(counter).zfill(6)+'.png'
            cv2.imwrite(name, frame,[cv2.IMWRITE_JPEG_QUALITY,90])
            counter += 1
                
        index+=1
        
    cap.release()

extract_frames_by_interval('example.avi', skip_frames=30)

这两种方法都可以有效地帮助开发者获取所需的静态画面用于后续处理或分析工作。具体选择哪种取决于应用场景和个人偏好。