使用Python和OpenCV实现动态背景的画中画效果

# 视频文件路径
input_video_path = 'sc/input_video.mp4'  # 输入视频文件路径
output_video_path = 'output_video_bg03.mp4'  # 输出视频文件路径
background_video_path = 'sc/bg_03b.mp4'  # MP4视频背景文件路径

# 背景透明度（0.0 完全透明，1.0 完全不透明）
background_opacity = 0.6

# 背景视频的最大帧数（如果需要截断背景视频）
max_background_frames = 1000

# 读取视频文件
cap = cv2.VideoCapture(input_video_path)  # 读取前景视频
background_cap = cv2.VideoCapture(background_video_path)  # 读取背景视频

# 获取视频帧的尺寸
ret, frame = cap.read()  # 读取第一帧
if not ret:
    print("无法读取视频文件")
    raise Exception("无法读取视频文件")  # 如果无法读取，抛出异常

height, width = frame.shape[:2]  # 获取帧的高和宽

# 定义黑色实心矩形的尺寸变量
top_bar_height = int(height * 0.30)  # 顶部黑色实心矩形的高度设置为视频总高度的30%
bottom_bar_height = int(height * 0.15)  # 底部黑色实心矩形的高度设置为视频总高度的15%

# 定义蒙版的位置和大小
mask_top_margin = top_bar_height  # 蒙版顶部距离
mask_bottom_margin = bottom_bar_height  # 蒙版底部距离
mask_height = height - mask_top_margin - mask_bottom_margin  # 蒙版高度
mask_width = width  # 蒙版宽度
mask_x = 0  # 蒙版中心x坐标
mask_y = mask_top_margin  # 蒙版中心y坐标

# 创建蒙版
mask = np.zeros((height, width, 3), dtype=np.uint8)  # 创建一个全黑的蒙版
cv2.rectangle(mask, (mask_x, mask_y), (mask_x + mask_width, mask_y + mask_height), (255, 255, 255), -1)  # 在蒙版中绘制一个白色矩形

# 定义视频写入器
fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # 定义视频编码格式
out = cv2.VideoWriter(output_video_path, fourcc, 20.0, (width, height))  # 创建视频写入对象

# 背景视频帧计数器
background_frame_index = 0

# 遍历视频帧
while cap.isOpened():
    ret, frame = cap.read()  # 读取前景视频的当前帧
    if not ret:
        break  # 如果读取结束，跳出循环

    # 在视频帧顶部绘制黑色实心矩形
    cv2.rectangle(frame, (0, 0), (width, top_bar_height), (0, 0, 0), -1)
    # 在视频帧底部绘制黑色实心矩形
    cv2.rectangle(frame, (0, height - bottom_bar_height), (width, height), (0, 0, 0), -1)
    # 将蒙版应用到帧上
    masked_frame = cv2.bitwise_and(frame, mask)
    # 读取背景视频的当前帧
    background_ret, background_frame = background_cap.read()
    if not background_ret or background_frame_index >= max_background_frames:
        # 如果背景视频读取失败或达到最大帧数，重新开始背景视频
        background_cap.set(cv2.CAP_PROP_POS_FRAMES, 0)
        background_frame_index = 0
        background_ret, background_frame = background_cap.read()

    # 确保背景帧的尺寸与视频帧匹配
    background_frame = cv2.resize(background_frame, (width, height))
    # 调整背景帧的透明度
    background_frame = background_frame.astype(float) * background_opacity
    background_frame = np.clip(background_frame, 0, 255).astype(np.uint8)

    # 创建反向蒙版
    inv_mask = cv2.bitwise_not(mask)
    # 将背景帧与蒙版的反向进行混合
    masked_background = cv2.bitwise_and(background_frame, inv_mask)
    # 将两个混合结果相加
    result_frame = cv2.add(masked_frame, masked_background)
    # 保存帧
    out.write(result_frame)
    # 更新背景视频帧索引
    background_frame_index += 1

# 释放资源
cap.release()
background_cap.release()
out.release()
cv2.destroyAllWindows()