用opencv拼接2个视频

import cv2
import numpy as np

video1 = cv2.VideoCapture('video1.mp4')
video2 = cv2.VideoCapture('video2.mp4')

fps = video1.get(cv2.CAP_PROP_FPS)
width = int(video1.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(video1.get(cv2.CAP_PROP_FRAME_HEIGHT))

output = cv2.VideoWriter('output.mp4', cv2.VideoWriter_fourcc(*'mp4v'), fps, (width, height))

while True:
    ret1, frame1 = video1.read()
    ret2, frame2 = video2.read()
    
    if not ret1 or not ret2:
        break
    
    # 拼接两个视频的帧
    combined_frame = np.hstack((frame1, frame2))
    
    # 写入输出视频文件
    output.write(combined_frame)

video1.release()
video2.release()
output.release()
cv2.destroyAllWindows()