视频逐帧保存

视频逐帧抽取。

以每一帧的时间戳做为文件名。

多线程。

将脚本文件所在的目录下视频文件抽帧，并以“视频文件名”作为存放帧的文件夹。

import cv2

import os

import sys

import threading

import tqdm

 

def extract_video_frame(src_video, dest_dir):

    assert os.path.splitext(src_video)[-1] == '.mp4', {src_video} + 'is not a video'

    meta_info = os.stat(src_video)

    ctime = meta_info.st_ctime

    #ctime_ns = meta_info.st_atime_ns

    print('create time: ', ctime)

    #print('ctime_ns: ',ctime_ns)

    #print('time: ', os.path.getctime(video_path))

    cap = cv2.VideoCapture(src_video)

    if os.path.exists(dest_dir) is False:

        os.makedirs(dest_dir, True)

 

    fps = cap.get(cv2.CAP_PROP_FPS)

    frame_count = cap.get(cv2.CAP_PROP_FRAME_COUNT)

    frame_interval = 1/fps

    timestamp = ctime

    assert cap.isOpened() == True, {src_video} + 'can not be opend'

    for i in tqdm.trange(int(frame_count), desc=src_video):

        ret, frame = cap.read()

   

        if ret:

            millseconds = cap.get(cv2.CAP_PROP_POS_MSEC)

            timestamp = ctime + millseconds

            name = dest_dir + '\\' + str(timestamp) + '.jpg'

            #name = os.path.join(dest_dir, str(timestamp) + '.jpg')

            cv2.imwrite(name, frame)

        else:

            print('finished!')

            break

    cap.release()

def run():

    print('current path is: ', sys.argv[0])

    print('current dir: ', os.getcwd())

    print('absolute dir: ', os.path.abspath('.'))

    input_dir = os.getcwd()

    video_list = []

    for video_name in os.listdir(input_dir):

        file = os.path.join(input_dir, video_name)

        if os.path.isfile(file):

            if os.path.splitext(file)[-1] == '.mp4':

                video_list.append(file)

    print(video_list)

    for video_name in video_list:

        threading.Thread(target = extract_video_frame, args = (video_name, os.path.splitext(video_name)[0])).start()

if __name__ == "__main__":

    run()

 

'''

param   define

cv2.VideoCapture.get(0)     CV_CAP_PROP_POS_MSEC     视频文件的当前位置（播放）以毫秒为单位

cv2.VideoCapture.get(1)     CV_CAP_PROP_POS_FRAMES 基于以0开始的被捕获或解码的帧索引

cv2.VideoCapture.get(2)     CV_CAP_PROP_POS_AVI_RATIO 视频文件的相对位置（播放）：0=电影开始，1=影片的结尾。

cv2.VideoCapture.get(3)     CV_CAP_PROP_FRAME_WIDTH 在视频流的帧的宽度

cv2.VideoCapture.get(4)     CV_CAP_PROP_FRAME_HEIGHT 在视频流的帧的高度

cv2.VideoCapture.get(5)     CV_CAP_PROP_FPS 帧速率

cv2.VideoCapture.get(6)     CV_CAP_PROP_FOURCC 编解码的4字-字符代码

cv2.VideoCapture.get(7)     CV_CAP_PROP_FRAME_COUNT 视频文件中的帧数

cv2.VideoCapture.get(8)     CV_CAP_PROP_FORMAT 返回对象的格式

cv2.VideoCapture.get(9)     CV_CAP_PROP_MODE 返回后端特定的值，该值指示当前捕获模式

cv2.VideoCapture.get(10)    CV_CAP_PROP_BRIGHTNESS 图像的亮度(仅适用于照相机)

cv2.VideoCapture.get(11)    CV_CAP_PROP_CONTRAST 图像的对比度(仅适用于照相机)

cv2.VideoCapture.get(12)    CV_CAP_PROP_SATURATION 图像的饱和度(仅适用于照相机)

cv2.VideoCapture.get(13)    CV_CAP_PROP_HUE 色调图像(仅适用于照相机)

cv2.VideoCapture.get(14)    CV_CAP_PROP_GAIN 图像增益(仅适用于照相机)（Gain在摄影中表示白平衡提升）

cv2.VideoCapture.get(15)    CV_CAP_PROP_EXPOSURE 曝光(仅适用于照相机)

cv2.VideoCapture.get(16)    CV_CAP_PROP_CONVERT_RGB 指示是否应将图像转换为RGB布尔标志

cv2.VideoCapture.get(17)    CV_CAP_PROP_WHITE_BALANCE × 暂时不支持

cv2.VideoCapture.get(18)    CV_CAP_PROP_RECTIFICATION 立体摄像机的矫正标注（目前只有DC1394 v.2.x后端支持这个功能）

'''