存在形状为(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文件系统的截图。