将视频随机抽取n帧

猫老壳

于 2019-09-17 11:36:35 发布

阅读量692

点赞数

分类专栏：视频分类文章标签：视频随机抽帧

本文链接：https://blog.csdn.net/weixin_42105432/article/details/100920916

版权

本文介绍如何从视频中随机抽取指定数量的帧，详细阐述了抽帧过程，包括选择关键帧的方法和实现步骤，对于视频处理和分析工作具有实际应用价值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import cv2
import PIL.Image as Image
import random
import numpy as np

def mp4_loader(filepath, nsample, seglen, mode):
    cap = cv2.VideoCapture(filepath)
    videolen = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
    sampledFrames = []
    for i in range(videolen):
        ret, frame = cap.read()
        # maybe first frame is empty
        if ret == False:
            continue
        img = frame[:, :, ::-1]
        sampledFrames.append(img)
    average_dur = int(len(sampledFrames) / nsample)
    imgs = []
    for i in range(nsample):
        idx = 0
        if mode == 'train':
            if average_dur >= seglen:
                idx = random.randint(0, average_dur - seglen)
                idx += i * average_dur
            elif average_dur >= 1:
                idx += i * average_dur
            else:
                idx = i
        else:
            if average_dur >= segle

最低0.47元/天解锁文章