在opencv中使用蒙版替换图像的一部分

import cv2
import numpy as np

# 读取原始图像和蒙版图像
original_image = cv2.imread('original_image.jpg')
mask_image = cv2.imread('mask_image.jpg', 0)  # 以灰度模式读取蒙版图像

# 读取要替换的图像部分
replacement_image = cv2.imread('replacement_image.jpg')

# 获取蒙版图像的尺寸
mask_height, mask_width = mask_image.shape

# 确保蒙版和原始图像大小一致
if original_image.shape[:2] != mask_image.shape:
    raise ValueError("Mask image must have the same dimensions as the original image")

# 使用蒙版进行图像替换
for c in range(0, 3):
    original_image[:, :, c] = np.where(mask_image == 255,
                                       replacement_image[:, :, c],
                                       original_image[:, :, c])

# 显示结果图像
cv2.imshow('Result Image', original_image)
cv2.waitKey(0)
cv2.destroyAllWindows()