python - 将所有非黑色像素转换为一种颜色不会产生预期的输出

Question

我试图选择非黑色像素，然后将它们着色为黑色，将黑色像素着色为白色。我用了solution Stack Overflow 上提供，但到目前为止它对我不起作用。

import numpy as np
import matplotlib.pyplot as plt


image = plt.imread('Perforated_carbon/faltu.png')
plt.imshow(image)
plt.show()

image_copy = image.copy()

black_pixels_mask = np.all(image == [0, 0, 0], axis=-1)

non_black_pixels_mask = ~black_pixels_mask
# or non_black_pixels_mask = np.any(image != [0, 0, 0], axis=-1)  

image_copy[black_pixels_mask] = [255, 255, 255]
image_copy[non_black_pixels_mask] = [0, 0, 0]

plt.imshow(image_copy)
plt.show()

这是我当前获得的图像

我最想要的是这个

其他信息:

>>> image.shape
(256, 192, 3)
>>> image.dtype
dtype('float32')
>>> import matplotlib; print(matplotlib.__version__)
2.0.0

Answer 1

改变一下怎么样

black_pixels_mask = np.all(image == [0, 0, 0], axis=-1); 

至

black_pixels_mask = np.all(image == [0, 0, 0], axis=2)