[PYTHON] Connect a large number of videos together!

Thing you want to do

Concatenate all videos in the directory into one video

Execution method

$ python3 combine_videos.py -i sample_videos/ -o output.mp4 
29.0 1280 720
Filename:  video01.mp4
Filename:  video02.mp4
Filename:  video03.mp4
Done.

combine_videos.py

#coding=utf-8
import os
import cv2
import glob
import argparse    # 1.Import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-i', '--input_dir', help='input video directory')
parser.add_argument('-o', '--output_video', help='output video filename')

args = parser.parse_args()

def combine_videos(directory, result):
    #Specify the video to input(Sort by string)
    videos_list = glob.glob('{}/*'.format(directory))
    videos_list.sort()

    #Specify MP4V as the format
    fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')

    #Acquisition of video information
    movie = cv2.VideoCapture(videos_list[0])
    fps    = movie.get(cv2.CAP_PROP_FPS)
    height = movie.get(cv2.CAP_PROP_FRAME_HEIGHT)
    width  = movie.get(cv2.CAP_PROP_FRAME_WIDTH)
    print(fps, int(width), int(height))

    #Open the output file
    out = cv2.VideoWriter(result, int(fourcc), fps, (int(width), int(height)))

    #Loading video
    for video in videos_list:
        movie = cv2.VideoCapture(video)
        print('Filename: ', video.split('/')[-1])
        while True:
            ret, frame = movie.read()
            out.write(frame)
            if not ret:
                break
    print('Done.')

if __name__ == '__main__':
    combine_videos(args.input_dir, args.output_video)