读取显示.npy类型的图像

import numpy as np
import matplotlib.pyplot as plt

# 读取 .npy 文件
image_data = np.load('image.npy')

# 显示图像的形状
print("图像的形状：", image_data.shape)

# 显示图像
plt.imshow(image_data[:, :, 0])  # 显示第一个通道的图像
plt.axis('off')  # 关闭坐标轴
plt.show()

图像的shape为(height, width, channels)，有时文件通道数不一定是0（灰度图像）或者3（彩色RGB图像），

当 channels不是0或者3时，若要显示图像，需要添加附加条件显示第几个通道，如下代码所示，此时才能正常显示

plt.imshow(image_data[:,:,0])  # 显示第一个通道的图像