cv2 - 用噪声屏蔽图像

Question

我有一张 cv2 形状为 (448,448,3) 的图像，我想屏蔽所有像素 rgb > 250 有噪音。我现在一个像素一个像素地做，但效率不高。 original masked。怎样才能一次完成这样的操作？

rand_mask = np.round(np.random.rand(448, 448, 3) * 255).astype(np.uint8)
    for i in range(0, 448):
        for j in range(0, 448):
                if arr[i, j, 0] and arr[i, j, 1] and arr[i, j, 2] > 250:
                    arr[i, j, :] = rand_mask[i, j, :]

Answer 1

idx = arr>250
arr[idx] = rand_mask[idx]