[英]replace mask with original image opencv Python
我正在尝试用原始图像像素替换我使用蒙版找到的对象。 我有一个蒙版,在未检测到物体的地方显示黑色,如果检测到,则显示为白色。 然后我在 where 语句中使用图像
image[np.where((image2 == [255,255,255].any(axis = 2))
我被困在这里,我不知道如何将找到的白色值更改为原始图像(与其他蒙版一起使用)。 我试过image.shape
但这没有用。
谢谢。
复制蒙版,然后从白色像素坐标在蒙版的白色像素上绘制原始图像。 您还可以检查mask == 255
以进行逐元素比较。 您不需要np.where,因为您可以通过mask == 255
创建的布尔掩码对数组进行索引。
out = mask.copy()
out[mask == 255] = original_image[mask == 255]
您可以使用按位运算。 尝试这个:
replaced_image = cv2.bitwise_and(original_image,original_image,mask = your_mask)
访问https://docs.opencv.org/3.3.0/d0/d86/tutorial_py_image_arithmetics.html进一步了解按位操作
import os
import cv2
from netpbmfile import imread
img_dir = '.'
mask_dir = '.'
new_bg = 'image.png'
def get_foreground(fg_image_name, mask_name, bg_image_name):
fg_image = cv2.imread(fg_image_name)
mask = imread(mask_name)
mask_inverse = (1-mask)
bg_image = cv2.imread(bg_image_name)
bg_image = cv2.resize(bg_image, (fg_image.shape[1], fg_image.shape[0]))
foregound = cv2.bitwise_and(fg_image, fg_image, mask=mask)
background = cv2.bitwise_and(bg_image, bg_image, mask=mask_inverse)
composite = foregound + background
return composite
image_fg = get_foreground(os.path.join(img_dir, "NP1_0.jpg"), os.path.join(mask_dir, "NP1_0_mask.pbm"), new_bg)
cv2.imwrite("foreground.jpg", image_fg)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.