数据标注员工作笔记：视频截图

Posted 2021-06-02 Mario cai

数据标注员工作笔记：视频截图

• • 视频分段
  • 视频合并
  • 视频按帧数截屏

视频分段

使用ffmpeg,在终端运行，先将位置调到代码和视频一起的位置

ffmpeg -ss 00:00:00 -t 00:01:45 -i s1.mp4 -vcodec copy -acodec copy outs1.mp4

视频合并

from moviepy.editor import *

video_list = ['outs5.mp4','outss5.mp4']

L = []
for video_path in video_list:
    video = VideoFileClip(video_path)
    L.append(video)

final_clip = concatenate_videoclips(L)
final_clip.to_videofile('./final5.mp4', fps=24, remove_temp=True)

视频按帧数截屏

import cv2
import argparse
import os


def parse_args():
    """
    Parse input arguments
    """
    parser = argparse.ArgumentParser(description='Process pic')
    parser.add_argument('--input', help='video to process', dest='input', default=None, type=str)
    parser.add_argument('--output', help='pic to store', dest='output', default=None, type=str)
    # default为间隔多少帧截取一张图片
    parser.add_argument('--skip_frame', dest='skip_frame', help='skip number of video', default=24, type=int)
    # input为输入视频的路径 ，output为输出存放图片的路径
    args = parser.parse_args(['--input',r'C:\\video_download\\final14.mp4' , '--output', r'C:\\video download'])
    return args


def process_video(i_video, o_video, num):
    cap = cv2.VideoCapture(i_video)
    num_frame = cap.get(cv2.CAP_PROP_FRAME_COUNT)
    expand_name = '.jpg'
    if not cap.isOpened():
        print("Please check the path.")
    cnt = 0
    count = 0
    while 1:
        ret, frame = cap.read()
        cnt += 1
        #  how
        # many
        # frame
        # to
        # cut
        if cnt % num == 0:
            count += 1
            cv2.imwrite(os.path.join(o_video, str(count) + expand_name), frame)

        if not ret:
            break


if __name__ == '__main__':
    args = parse_args()
    if not os.path.exists(args.output):
        os.makedirs(args.output)
    print('Called with args:')
    print(args)
    process_video(args.input, args.output, args.skip_frame)

以上全都借鉴来自CSDN的某些大佬，但是具体地址忘记了。
小弟萌新一个，多多包涵.