我有一个灰度图像 - 实际上是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]
答案 0 :(得分:1)
如果A是您的矩阵,则二进制矩阵B为:
B = np.where(A <= 240,1,0)