将图片变成黑色或白色。怎么样？

Question

我想编写一个函数来将我的RGB图片更改为黑色或白色，而不是灰度。我应该只为此使用NumPy函数。我希望低于特定阈值的RGB变为黑色[0 0 0]，高于特定阈值的白色[255 255 255]。如何使图像的每个像素向上或向下取整？dtype ='uint8'。我通过执行以下操作将图片上传到python：img = mpimg.imread('Picture.png')

还有另一个问题，如何编写将一半像素变成黑色，一半像素变成白色的函数？ （使用numpy.median）

Answer 1

a = np.random.rand(64,64,3)*255

threshold = 100
b = (a > threshold)*255

您可以在numpy数组上使用诸如>之类的运算符来生成布尔数组，该布尔数组指示条件是对哪些单元格为真还是假（阈值）。然后，您可以将此布尔数组乘以所需的阈值结果（在本例中为255和0）。