图像修复的数据集制作

Question

如何实现将不规则mask添加到原图上，得到有缺失的图像呢，类似于将1.2结合得到3，Python新手上路，求各位指教一下，万分感谢！

Answer 1

你可以使用OpenCV库来实现将不规则mask添加到原图上的操作。

首先读取原图和mask图片，并使用cv2.bitwise_and()函数将mask图片与原图进行位运算，得到一张有缺失部分的图像。

具体代码如下：

import cv2

# 读取原图和mask图片
original_image = cv2.imread("original.jpg")
mask_image = cv2.imread("mask.jpg")

# 将mask图片与原图进行位运算
result_image = cv2.bitwise_and(original_image, mask_image)

# 保存结果图片
cv2.imwrite("result.jpg", result_image)

如果你想要将缺失部分填充为白色，可以使用cv2.bitwise_not()函数将mask图片取反，然后使用cv2.bitwise_and()函数将取反后的mask图片与原图进行位运算，得到一张有白色缺失部分的图像。

具体代码如下：

import cv2

# 读取原图和mask图片
original_image = cv2.imread("original.jpg")
mask_image = cv2.imread("mask.jpg")

# 将mask图片取反
mask_image_inverted = cv2.bitwise_not(mask_image)

# 将取反后的mask图片与原图进行位运算
result_image = cv2.bitwise_and(original_image, mask_image_inverted)

# 保存结果图片
cv2.imwrite("result.jpg", result_image)

希望这些信息能够帮助你。