python - 如何将彩色像素更改为白色像素

Question

我有一个彩色图像作为输入，我想检查方差的颜色信息(如 [0, 0, 0] - [255, 255, 255])。因此，如果方差超过某个点，我想将其更改为白色。

所以类似的事情:

for y in range(img.shape[0]):
    for x in range(img.shape[1]):
        if numpy.var(img[y][x]) > 1000:
            img[y][x] = [255, 255, 255]

但我需要良好的表现。所以我尝试使用 numpy.where() 函数，但找不到解决方案。

Answer 1

您可以使用 numpy 的索引来实现此目的:

import numpy as np
import matplotlib.pyplot as plt

img = (np.random.rand(100,100,3)*255).astype(int)

img2 = np.copy(img)
img2[np.var(img, 2)>1000] = np.array([255, 255, 255])

fig, ax = plt.subplots(ncols=2)

ax[0].imshow(img)
ax[1].imshow(img2)

np.var的第二个参数是要计算方差的轴；在本例中是颜色。

结果: