读取图片

import matplotlib.pyplot as plt  # plt 用于显示图片
import matplotlib.image as mpimg  # mpimg 用于读取图片

lena = mpimg.imread(‘index.jpg‘)  # 读取和代码处于同一目录下的 lena.png
# 此时 index 就已经是一个 np.array 了，可以对它进行任意处理
print(lena.shape)  # (512, 512, 3)
print(lena)
plt.imshow(lena)  # 显示图片
plt.axis(‘off‘)  # 不显示坐标轴
plt.show()

# 显示图片的第一个通道
lena_1 = lena[:,:,0]
# lena_1 = lena[:,:,1]
# lena_1 = lena[:,:,2]
plt.imshow(lena_1)
plt.show()

# 显示灰度图
# 方法1
plt.imshow(lena_1, cmap=‘Greys_r‘)
plt.show()
# 方法2
img = plt.imshow(lena_1)
img.set_cmap(‘gray‘) # ‘hot‘ 是热量图
plt.show()