如何仅使用 numpy 屏蔽图像

Question

为什么这个遮罩层不遮罩图像。

import matplotlib.image as mpimg
import numpy
path = 'inp.jpg'
arr = numpy.array(Image.open(path))
 img = mpimg.imread(path)
 black_pixels_mask = np.all(img == [0, 0, 0], axis=-1)
 img[black_pixels_mask] = [255,255,255]

结果 img 应该是屏蔽的 one.and 代码应该将黑色替换为白色。就像一个样本颜色。

Answer 1

试试这个代码

arr[numpy.all(arr == [0, 0, 0], axis=-1)]=[255,255,255]
data = Image.fromarray(arr)
data.save(path)