import numpy as np
import matplotlib.pyplot as plt
# 读取图像
n1 = plt.imread('保险图.png') / 255.0 # 将像素归一化到0-1范围,因为像素通常在0~255之间
# 确保图像是RGB格式
if n1.ndim == 3 and n1.shape[2] == 4: # 如果图像是RGBA,只取RGB部分
n1 = n1[:, :, :3]
# 这段代码检查 n1 是否是一个三维数组,
# 并且最后一个维度的大小是否为 4,
# 这意味着图像可能是 RGBA 格式(红、绿、蓝、透明度)。
# 如果是 RGBA 格式,
# n1[:, :, :3] 将只保留 RGB 三个通道
# ,丢弃透明度通道
# 创建灰度公式的权重数组,以下数字是固定的
n2 = np.array([0.299, 0.587, 0.114])
# 正确扩展n2的维度,使其可以与n1进行逐元素乘积
n2_expanded = n2[np.newaxis, np.newaxis, :]
# 逐元素乘积后求和,得到灰度图像
n1_gray = np.sum(n1 * n2_expanded, axis=2)
# 显示灰度图像
plt.imshow(n1_gray, cmap='gray')
# 显示图像
plt.show()
python(模块+numpy-数组,矩阵)
最新推荐文章于 2024-09-09 00:00:00 发布