cv2 - 用噪聲屏蔽圖像

Question

我有一個形狀為 (448,448,3) 的cv2圖像，並且想要用噪聲掩蓋所有像素 rgb > 250。 我現在正在逐個像素地進行處理，但效率不高。 原來蒙面。 我怎樣才能一次完成這樣的操作？

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]