如何在Scipy中将图像加载为具有阈值的二进制图像?

时间:2016-04-21 17:20:59

标签: python numpy scipy

我有一个灰度图像 - 实际上是MNIST数据集 - 我需要将其转换为阈值为240的二进制图像,以便240以下的所有值都是1,而上面的所有值都是零。

这是matlab中的一个函数,所以我确定在scipy中有一些相应的函数...但是它正在逃避我的搜索。

同样,如果我有一个(60000,28,28)形状的ndarray,如何有条件地检查所有值并将值设置在240以上,其余值设置为1?

在伪numpy代码中,

image_array = big_array_of_28x28_images

bw_image_array = image_array[image_array > 240 yield 0, else yield 1]

1 个答案:

答案 0 :(得分:1)

如果A是您的矩阵,则二进制矩阵B为:

B = np.where(A <= 240,1,0)