opencv将视频转成图片

最新推荐文章于 2024-09-03 17:03:19 发布

原创

最新推荐文章于 2024-09-03 17:03:19 发布 · 1.3k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#图像处理 #opencv #视频转图片

本文介绍了一种使用Python和OpenCV从视频中提取图片的方法，以供图像处理与深度学习任务使用。通过设定时间间隔和图片尺寸，可以高效地从大量视频数据中获取训练所需的图像。

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

最近在做图像处理与深度学习的实践，获得了视频数据，我要将它每隔几帧提取一张图片，然后对图片进行标注，下面分享一下视频提取图片的代码，使用python。

import os
import cv2

videos_src_path = "./video/"
# video_formats = [".MP4", ".MOV"]          我的数据集都是.mp4所以不需要进行分类判断
frames_save_path = "./image/"
width = 2048
height = 1536
time_interval = 5


def list_all_files(rootdir):
    import os
    _files = []
    list = os.listdir(rootdir) #列出文件夹下所有的目录与文件
    for i in range(0,len(list)):
           path = os.path.join(rootdir,list[i])
           if os.path.isdir(path):
              _files.extend(list_all_files(path))
           if os.path.isfile(path):
              _files.append(path)
    return _files

def video2frame(video_src_path, frame_save_path, frame_width, frame_height, interval):
    """
    将视频按固定间隔读取写入图片
    :param video_src_path: 视频存放路径
    :param formats:　包含的所有视频格式
    :param frame_save_path:　保存路径
    :param frame_width:　保存帧宽
    :param frame_height:　保存帧高
    :param interval:　保存帧间隔
    :return:　帧图片