OpenCV:Python相当于C++中的“setTo”

OpenCV : Python equivalent of `setTo` in C++

本文关键字：setTo 中的 C++ Python 相当于 OpenCV      更新时间：2023-10-16
在

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
你可以

简单地使用img[mask > 0] = 0，这是python等价物img.setTo(0, mask);

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]添加一个维度，然后乘以图像。

顺便说一句，numpy.newaxis与使用mask[:,:,None]相同。

相关文章：