问题原因拆解 #

你遇到的ValueError: invalid palette size本质是PIL自动生成的P模式位数和你传入的调色板条目数不匹配：

• 当你执行Image.fromarray(labelMap).convert('P')时，PIL会自动扫描labelMap中的像素值范围，生成一个「刚好够用」的调色板位数。比如如果你的标签数组里最大的索引值是63，PIL会默认用6位P模式（对应最多64个调色板条目，总长度就是64×3=192字节）。
• 这时候你传入768字节的调色板（对应256个条目），就超出了当前P模式允许的最大条目数，自然触发报错。而当调色板长度小于192时，刚好匹配PIL自动生成的低位数模式，所以能正常运行。

解决办法 #

有两种可靠的方式解决这个问题，根据你的需求选择：

1. 强制转换为8位P模式（推荐） #

直接让PIL生成支持256个条目的8位P模式，这样768字节的调色板就完全匹配了。你可以用两种写法：

写法一：先转灰度再转P模式

# 先将数组转为8位灰度图，再转成8位P模式，确保调色板支持256个条目
png = Image.fromarray(labelMap).convert('L').convert('P')
png.putpalette(cmap)
png.save(pngPath, format='PNG')

写法二：直接指定convert参数

# 转换时强制使用256色的调色板，生成8位P模式
png = Image.fromarray(labelMap).convert('P', colors=256)
png.putpalette(cmap)
png.save(pngPath, format='PNG')

2. 匹配调色板条目数与图像最大索引 #

如果你的标签数组里的索引值范围很小（比如最大索引是99），可以只生成对应长度的调色板（比如3×100=300字节），这样也能避免报错。不过这种方法灵活性较差，不如第一种通用。

验证方法 #

你可以在转换后打印图像的模式和当前调色板长度，确认是否是8位P模式：

print(png.mode)  # 应该输出 'P'
print(len(png.getpalette()))  # 8位P模式默认调色板长度是768

内容的提问来源于stack exchange，提问作者梁生珺