CV | Python + OpenCV从视频中提取“指定帧”

简介

补充：很多人说这个跟“关键帧”没有关系，我不知道“关键帧”这个次是否是特指的，为了避免大家误解，此处改为：“指定帧”。

其实已经写过两篇关于从视频中提取“指定帧”的博客了（CV | OpenCV 提取AVI视频中“指定帧”图像（修改版） ，CV | 使用Python+OpenCV提取avi和mp4格式视频中的“指定帧”）。但是今天把代码又修改完善了一下，感觉比之前的几个版本都好（哈哈自夸中。。。），因此再次贴出利用OpenCV从视频中提取“指定帧”的Python代码。

程序

功能：自动的读取指定文件夹下的所有文件，并且按照文件类型和名称的不同，提取该文件夹下所有视频，并按照名称的不同，分类保存从视频中提取的“指定帧”。

import cv2
import os

# 从.avi 类型的视频中提取图像
def splitFrames(sourceFileName):

    # 在这里把后缀接上
    video_path = os.path.join('./data/video_dataset/', sourceFileName + '.avi')
    outPutDirName = './data/video_dataset/video/' + sourceFileName + '/'

    if not os.path.exists(outPutDirName):
        #如果文件目录不存在则创建目录
        os.makedirs(outPutDirName)

    cap = cv2. VideoCapture(video_path) # 打开视频文件
    num = 1
    while True:
        # success 表示是否成功，data是当前帧的图像数据；.read读取一帧图像，移动到下一帧
        success, data = cap.read()
        if not success:
            break
        # im = Image.fromarray(data, mode='RGB') # 重建图像
        # im.save('C:/Users/Taozi/Desktop/2019.04.30/' +str(num)+".jpg") # 保存当前帧的静态图像
        cv2.imwrite( outPutDirName +str(num)+".jpg", data)

        num = num + 1

        # if num % 20 == 0:
        #     cv2.imwrite('./Video_dataset/figures/' + str(num) + ".jpg", data)

        print(num)
    cap.release()

# 从.mp4 数据类型的视频中提取图像
def splitFrames_mp4(sourceFileName):

    # 在这里把后缀接上
    video_path = os.path.join('./data/video_dataset/', sourceFileName + '.mp4')
    times = 0
    # 提取视频的频率，每25帧提取一个
    # frameFrequency = 25
    # 输出图片到当前目录vedio文件夹下
    outPutDirName = './data/video_dataset/video/' + sourceFileName + '/'

    # 如果文件目录不存在则创建目录
    if not os.path.exists(outPutDirName):
        os.makedirs(outPutDirName)

    camera = cv2.VideoCapture(video_path)
    while True:
        times+=1
        res, image = camera.read()
        if not res:
            # print('not res , not image')
            break

        # if times%frameFrequency==0:
        #     cv2.imwrite(outPutDirName + str(times)+'.jpg', image)
        #     print(outPutDirName + str(times)+'.jpg')

        cv2.imwrite(outPutDirName + str(times) + '.jpg', image)
        print(times)
    print('图片提取结束')
    camera.release()

if __name__ == '__main__':

    im_file = './data/Video_dataset/'

    # for im_name in im_names:
    for im_name in os.listdir(im_file):
        suffix_file = os.path.splitext(im_name)[-1]
        if suffix_file == '.mp4':
            print('~~~~~~~~~~ 从.mp4 视频提取图像 ~~~~~~~~~~~~~~~')

            sourceFileName = os.path.splitext(im_name)[0]
            splitFrames_mp4(sourceFileName)

        elif suffix_file == '.avi' :
            print('~~~~~~~~~~ 从.avi 视频提取图像 ~~~~~~~~~~~~~~~')

            sourceFileName = os.path.splitext(im_name)[0]
            splitFrames(sourceFileName)