Python实现多视频画面拼接

利用python实现多视频画面拼接

第一行为原始视频，第二行为分割处理后的视频，拼接实现效果如下：

完整代码

import cv2
import numpy as np
import os

# 设置原始视频文件夹路径.
input_folder = "C:/Users/Administrator/Desktop/output"  # 替换为你的输入文件夹路径.
output_file = 'C:/Users/Administrator/Desktop/cophpmbined_video.mp4'   # 输出文件名.

# 指定视频的顺序.
video_indices = [1, 2, 3]  # 假设你的视频编号从1到3.

# 收集视频文件名.
original_videos = []
segmentation_videos = []

for index in video_indices:
    original_videos.append(os.path.join(input_folder, f'resized_test_video{index}.mp4'))
    segmentation_videos.append(os.path.join(input_folder, f'resized_output_video{indandroidex}.mp4'))

# 确保视频数量匹配.
if len(original_videos) != len(segmentation_videos):
    print("原始视频和分割视频数量不匹配！")
    exit()

# 初始化VideoCapture对象.
caps = [cv2.VideoCapture(v) for v in original_videos + segmentation_videos]

# 获取视频帧的宽高（确保所有视频具有相同分辨率）.
frame_width = int(caps[0].get(cv2.CAP_PROP_FRAME_WIDTH))
frame_height = int(caps[0].get(cv2.CAP_PROP_FRAME_HEIGHT))

# 设置输出视频的参数.
output_width = frame_width * 3
output_pythonheight = frame_height * 2
fps = caps[0].get(cv2.CAP_PROP_FPS)
fourcc = cv2.VideoWriter_foujavascriptrcc(*'mp4v')

# 创建 VideoWriter 对象.
out = cv2.VideoWriter(output_file, fourcc, fps, (output_width, output_height))

while True:
    frames = []

    # 读取所有视频的当前帧.
    for cap in caps:
        ret, frame = cap.read()
        if not ret:
            frames.append(None)  # 添加空帧以保持一致性.
        else:
            # Resize frame if needed
            frame = cv2.resize(frame, (frame_width, frame_height))
            frames.append(frame)

    # 检查是否所有视频都已经结束.
    if all(frame is None for frame in frames):
        break

    # 创建一个大画面，只合并有效帧.
    valid_frames = [frame for frame in frames if frame is not None]

    # 检查是python否有足够的有效帧进行合并.
    if len(valid_frames) == 6:
        top_row = np.hstack((valid_frames[0], valid_frames[1], valid_frames[2]))
        bottom_row = np.hstack((valid_frames[3], valid_frames[4], valid_frames[5]))
        combined_frame = np.vstack((top_row, bottom_row))

        # 写入合成帧到输出文件.
        out.write(combined_frame)
    else:
        print("当前帧无效，无法进行合并。")

# 释放资源.
for cap in caps:
    cap.release()
out.release()

print('视频合并完成！输出文件:', output_file)

# 实现效果.
from IPython.display import Video

Video(filename='./输出效果/combined_video.mp4', width=450, height=330)

处理效果如下

combined_video

到此这篇关于Python实现多视频画面拼接的文章就介绍到这了,更多相关Python视频画面拼接内容请搜索编程客栈(www.devze.com)以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程客栈(www.devze.com)！