现在已经有语义分割的掩码图了，帮我写一个语义分割之后的mask图从原图中提取出来并且保持和原图颜色一样，并保存图片的python代码，

当你有了语义分割的掩码图后，你可以使用该掩码图从原图中提取出对应的区域，并保持相同的颜色。下面是一个示例的 Python 代码，可以完成这个任务：

import cv2
import numpy as np

# 读取原图和掩码图
original_image = cv2.imread('original_image.jpg')
mask_image = cv2.imread('mask_image.jpg', 0)  # 读取为灰度图像

# 创建一个与原图相同大小的彩色掩码图像
color_mask = np.zeros_like(original_image)

# 提取掩码图中非零像素对应的原图像素，并赋值给彩色掩码图
color_mask[mask_image != 0] = original_image[mask_image != 0]

# 保存彩色掩码图
cv2.imwrite('color_mask.jpg', color_mask)

请确保将 `original_image.jpg` 替换为你的原始图像文件名，将 `mask_image.jpg` 替换为你的语义分割掩码图像文件名。最后，将生成的彩色掩码图保存为 `color_mask.jpg`。

这段代码使用 OpenCV (`cv2`) 库来处理图像。它首先读取原始图像和语义分割的掩码图像，然后创建一个与原图相同大小的空白彩色掩码图像。接下来，它通过比较掩码图中非零像素的位置，将对应的原图像素值赋给彩色掩码图。最后，将彩色掩码图保存到文件中。

请确保安装了 OpenCV 库，可以使用 `pip install opencv-python` 来安装。