Question

C ++中Mat::setTo的python等价物是什么？

我尝试做的是通过掩码设置值：

Mat img;
...
img.setTo(0, mask);

更新

这是可能的解决方案：

#set by mask area to zero
img= np.random.rand(200, 200, 3) * 255
img= img.astype(np.uint8)
mask = np.zeros((200, 200), np.uint8)
mask[10:100, 60:140] = 255
inv_mask= cv2.bitwise_not(mask)
n_channels= img.shape[2]
for i in range(0,n_channels):
    img[..., i]= img[..., i] * (inv_mask/255)

#to set arbitary value
img= np.random.rand(200, 200, 3) * 255
img= img.astype(np.uint8)
mask= np.zeros((200,200), np.uint8)
mask[10:100, 60:140]= 255
mask_bool= np.where(mask > 0)
value= 120
img[mask_bool]= value

Answer 1

你可以简单地使用img[mask > 0] = 0，它是img.setTo(0, mask);

的python等价物

Answer 2

python中的opencv使用numpy作为后端（更具体地说是numpy.ndarray）用于图像。这意味着您可以执行以下操作：

import numpy
img = numpy.zeros((256, 256, 3))  # This initializes a color image as an array full of zeros

然后，如果我希望将第三个频道设置为1，我所要做的就是：

img[:, :, 2:3] = 1

因此，如果Mask具有与我的第三个通道相同的尺寸，我可以使用它来设置第三个通道：

img[:, :, 2:3] = mask

评论后编辑：如果您只是希望使用蒙版将图像像素设置为零，则简单image * mask就足够了。确保蒙版的尺寸与图像相同。

编辑＃2：是的，这可能会有效，但如果你的面具的形状是唯一的问题，我会使用mask[:, :, numpy.newaxis]添加尺寸，然后与图像相乘。

btw numpy.newaxis与使用mask[:,:,None]相同。

OpenCV：Python相当于C ++中的`setTo`

2 个答案: