有没有一种聪明的方法可以通过像素值更改代码来简化我的像素？

Question

# masking 
masked = mask.copy()
for py in range(0,h):
    for px in range(0,w):
        if masked[py][px][0] != 0:
            masked[py][px] = im[py][px]
masked = cv2.cvtColor(masked, cv2.COLOR_RGB2BGR)

有什么奇特的方法可以将我的脏代码转换为带有 cv2 或 numpy 函数的简单代码吗？

Answer 1

我认为这可以帮助替换循环（与您的循环非常相似）：

indices = np.where(masked[:,:,0] != 0)
masked[indices]=im[indices]

有没有一种聪明的方法可以通过像素值更改代码来简化我的像素？

Is there a smart way to simplipy my pixel by pixel value changing code?

numpy

cv2