关于在图像中显示和保存0-1个ndarray

时间:2016-11-21 17:37:39

标签: python numpy image-processing matplotlib scipy

存在形状为(5508,1,64,80)的ndarray,其存储对应于5508个图像的5508二维0-1阵列。我正在尝试使用以下代码段将它们保存到5508个图像中。

import numpy as np
import os
image_path = "/data/image_test"

import matplotlib.pyplot as plt
import matplotlib.cm as cmp

if __name__ == '__main__':
    img_array = np.load('/data/imgs_mask_test.npy')
    print(img_array.shape)
    for i in range(img_array.shape[0]):
        file_name = os.path.join(image_path,'_'+str(i)+'.png')
        plt.imsave(file_name,img_array[i,0],cmap=cmp.gray)

我不确定此代码段是否正确。我检查了生成的图像,这些图像都是黑色的。这是否意味着原始ndarray中的所有值都是0?从numpy的角度来看,我们可以获得入口值为1的所有位置吗?

此外,我怀疑是否需要将此0-1重新缩放为0-255?但在我看来它仍然会给我两个值,例如0和255。

以下是Windows文件系统的截图。

enter image description here

0 个答案:

没有答案