视频分帧，并保存帧图片

写在前边

要解决的问题：针对N个视频，将将视频按照帧来裁剪，并且按照自己想要的帧间隔保存相应的帧图片。

使用的工具：python、pycharm

博主：代码界的新手

创新点：进行了修改，适合自己的模型。

代码

代码展示：

#coding=gbk
import cv2
import os


def splitFrames(sourceFileName):

    # 在这里把后缀接上
    video_path = os.path.join('database', sourceFileName + '.avi')
    outPutDirName = 'database' + 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
        cv2.imwrite(outPutDirName +str(num)+".jpg", data)

        num = num + 1

        print(num)
    cap.release()

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

    # 在这里把后缀接上
    video_path = os.path.join('database', sourceFileName + '.mp4')
    times = 0
    # 提取视频的频率，25帧提取一个每
    # frameFrequency = 25
    # 输出图片到当前目录vedio文件夹下
    outPutDirName ='database' + 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:
            break
        #cv2.imwrite(outPutDirName + str(times) + '.jpg', image)#输出所有图片帧
        if times % 5 == 0:
            cv2.imwrite(outPutDirName + str(times) + '.jpg', image)
            print(times)
    print('图片提取结束')
    camera.release()

if __name__ == '__main__':

    im_file = 'database'

    # 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)