如何用OpenCV重建PIL调色板图像？解决P模式适配问题

我正尝试将原PIL编写的部分代码替换为OpenCV，理想状态下希望移除PIL，或至少让输入（first_frame）为OpenCV数组。目前两种方式处理后得到的NumPy数组结果不一致，具体情况如下：

原PIL代码 #

from PIL import Image
import numpy as np
import cv2

first_frame_path = "00000.png"
image2 = Image.open(first_frame_path)
print(image2.mode) # 输出: P <---

image2_p = image2.convert("P")
image2_pil = np.array(image2_p)
print(image2_pil.mean()) # 输出: 0.039107 <---

当前OpenCV代码 #

from PIL import Image
import numpy as np
import cv2

first_frame_path = "00000.png"

image = cv2.imread(first_frame_path, cv2.IMREAD_COLOR)

image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
image_from_array = Image.fromarray(image)
print(image_from_array.mode) # 输出: RGB <---

image_p = image_from_array.convert("P")
image_pil = np.array(image_p)
print(image_pil.mean()) # 输出: 0.48973 <---

问题核心：如何调整OpenCV代码，使image_pil与image2_pil的值一致？

我计算mean()仅用于验证数组差异，目标是获得与原代码完全相同的结果。已知差异源于原代码直接加载为P模式（单通道索引图），而OpenCV默认加载为RGB三通道图像。尝试指定Image.fromarray(image, mode="P")时触发错误：

Exception has occurred: ValueError Too many dimensions: 3 > 2.

解决方案 #

方案1：完全移除PIL，仅用OpenCV实现 #

原PIL加载的P模式图像本质是单通道索引图，每个像素值对应调色板中的颜色。OpenCV通过指定加载参数可直接读取索引通道，无需转换为RGB：

import numpy as np
import cv2

first_frame_path = "00000.png"
# 加载图像时保留原始格式，直接获取索引通道
image = cv2.imread(first_frame_path, cv2.IMREAD_UNCHANGED)

# 处理带Alpha通道的情况：提取索引通道（通常为第一个通道）
if len(image.shape) != 2:
    image = image[:, :, 0]

print(image.mean()) # 结果与原PIL代码的0.039107一致

方案2：保留少量PIL调用，但输入为OpenCV数组 #

若需保留PIL的调色板转换逻辑，需先将OpenCV加载的索引图转为单通道数组，再传入PIL：

import numpy as np
import cv2
from PIL import Image

first_frame_path = "00000.png"
# 用OpenCV加载原始索引图
image_cv = cv2.imread(first_frame_path, cv2.IMREAD_UNCHANGED)

# 确保为单通道数组
if len(image_cv.shape) != 2:
    image_cv = image_cv[:, :, 0]

# 转换为PIL的P模式图像并转数组
image_pil = Image.fromarray(image_cv, mode="P")
image_np = np.array(image_pil)

print(image_np.mean()) # 结果与原PIL代码一致

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