python 合成视频段

有多个视频，为了测试方便，从每个视频中取一段合成为一个视频。

import cv2
import argparse

def parse_args():
    args = argparse.ArgumentParser()
    args.add_argument('videos',nargs='+')
    return args.parse_args()

def videos2img(videos):
	fps = 25
	size = (1920, 1080)
	fourcc = cv2.VideoWriter_fourcc(*'MJPG')
	videoWriter = cv2.VideoWriter('concat.mp4', fourcc, fps, size)

	for video in videos:
		videoCapture = cv2.VideoCapture(video)
	
		count = 0
		success, frame = videoCapture.read()
		while success:
			if count < 300:
				videoWriter.write(frame)
				# cv2.imshow("winname", frame)
				# cv2.waitKey()
				success, frame = videoCapture.read()
				if frame.shape[1] < 1920:
					frame = cv2.resize(frame, size, interpolation=cv2.INTER_CUBIC)
					# cv2.imshow("winname", frame)
					# cv2.waitKey()
				count += 1
			else:
				break

if __name__ == "__main__":
	args = parse_args()
	videos2img(args.videos)