使用matplotlib.pyplot保存二值掩码为何生成4通道PNG？如何解决？

问题原因解析 #

你遇到的(2,2,4)形状是因为matplotlib的plt.imsave默认会把图像保存为RGBA格式——也就是红、绿、蓝、透明度四个通道。哪怕你指定了灰度colormap，函数的默认行为还是会添加alpha通道。你的布尔数组会先被转换成0（False）和1（True）的浮点数，再通过灰度colormap映射后，被打包进RGBA的四个通道中，所以读取后就出现了4通道的结果。

额外提一句：就算你保存的是纯灰度图，用plt.imread读取时，默认也会把单通道的PNG转换成RGB三通道（三个通道值完全相同），这也是matplotlib的默认处理逻辑。

解决方案：保存为单通道0/1灰度图 #

下面两种方法可以帮你实现需求，优先推荐第一种，更直接可靠：

方法1：用Pillow（PIL）库直接处理

Pillow可以直接创建并保存单通道的灰度图，完全避开多余的通道问题：

import numpy as np
from PIL import Image

# 你的二值掩码数组
mask = np.array([[False, True], [False, True]])
# 把布尔值转换成0/255的8位整数（PNG灰度图用0-255表示黑/白）
gray_mask = (mask * 255).astype(np.uint8)
# 以8位灰度模式（mode='L'）保存
Image.fromarray(gray_mask, mode='L').save("single_channel_mask.png")

# 读取验证
img = Image.open("single_channel_mask.png")
print(np.array(img).shape)  # 输出：(2, 2)

这里的mode='L'代表8位单通道灰度图，保存后文件就是纯单通道的，读取后直接得到你想要的二维数组形状。

方法2：调整matplotlib参数（需注意读取方式）

如果你一定要用matplotlib，需要手动关闭alpha通道，并且后续读取时要用灰度模式读取：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import cv2  # 用于灰度模式读取

mask = np.array([[False, True], [False, True]])
# 把布尔数组转成0/255的8位整数，关闭透明度
plt.imsave("mask_gray.png", (mask*255).astype(np.uint8), 
           cmap='gray', format='png', transparent=False)

# 注意：用plt.imread读取还是会得到3通道，所以要用OpenCV的灰度模式读取
img = cv2.imread("mask_gray.png", cv2.IMREAD_GRAYSCALE)
print(img.shape)  # 输出：(2, 2)

这种方法需要注意读取工具的选择，因为matplotlib自身的imread会自动扩展通道数，所以得用OpenCV或Pillow来读取单通道图像。

内容的提问来源于stack exchange，提问作者TheDude