如何将图像的遮罩部分叠加到新图像上？

import cv2

def overlay_mask(original_image, mask_image):
    # 确保图像尺寸相同
    mask_image = cv2.resize(mask_image, (original_image.shape[1], original_image.shape[0]))

    # 将图像转换为像素矩阵
    original_pixels = original_image.astype(float)
    mask_pixels = mask_image.astype(float) / 255.0

    # 遍历遮罩图像的每个像素
    for i in range(original_image.shape[0]):
        for j in range(original_image.shape[1]):
            # 检查遮罩图像的像素值
            if mask_pixels[i, j] > 0:
                # 将原始图像的像素值复制到新图像
                original_pixels[i, j] = original_pixels[i, j] * (1 - mask_pixels[i, j]) + mask_pixels[i, j]

    # 将像素矩阵转换回图像
    new_image = original_pixels.astype(int)

    return new_image

# 加载原始图像和遮罩图像
original_image = cv2.imread('original_image.jpg')
mask_image = cv2.imread('mask_image.jpg', 0)  # 以灰度模式加载遮罩图像

# 将遮罩部分叠加到原始图像上
result_image = overlay_mask(original_image, mask_image)

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