Python中OpenCV加载4通道CMYK NumPy数组崩溃的问题咨询

用户提供的测试代码：

import numpy as np
from PIL import Image
import cv2

with Image.open("image.png") as im:
   
    im = im.convert("CMYK")# not a true CMYK conversion here  
    im.show(title="image")
    img = np.array(im)
    #cv2.imshow('image', img)

问题描述 #

我希望使用OpenCV查看CMYK格式文件，并读取CMYK色彩空间的像素值。我尝试加载图片后将其转为4通道CMYK格式，再用OpenCV查看，但注释掉了cv2.imshow代码，因为这会导致Python崩溃，甚至需要重新安装OpenCV。请问OpenCV是否支持查看形状为(x,x,0:3)的uint8类型NumPy数组？若支持，请给出相关代码。

解答 #

OpenCV的imshow支持显示形状为(H,W,3)的uint8数组（即3通道图像），但前提是数组对应的色彩空间符合OpenCV的预期（比如BGR、RGB、灰度等）。直接显示4通道CMYK数组会崩溃，是因为OpenCV默认将4通道解析为RGBA，而CMYK的通道含义与RGBA完全不同，导致渲染错误。

以下是可行的代码示例：

1. 单独查看CMYK前3通道或单个通道

import numpy as np
from PIL import Image
import cv2

# 通过PIL加载图片并转换为CMYK格式
with Image.open("image.png") as im:
    im_cmyk = im.convert("CMYK")
    img_cmyk = np.array(im_cmyk)  # 得到形状为(H, W, 4)的uint8数组

# 提取CMYK的前3个通道（青、品红、黄）
img_cmy = img_cmyk[..., :3]

# 显示前3通道图像
cv2.imshow('CMY Channels', img_cmy)
cv2.waitKey(0)
cv2.destroyAllWindows()

# 也可以单独显示每个通道的灰度图
c_channel = img_cmyk[..., 0]  # 青通道
m_channel = img_cmyk[..., 1]  # 品红通道
y_channel = img_cmyk[..., 2]  # 黄通道
k_channel = img_cmyk[..., 3]  # 黑通道

cv2.imshow('Cyan', c_channel)
cv2.imshow('Magenta', m_channel)
cv2.imshow('Yellow', y_channel)
cv2.imshow('Key (Black)', k_channel)
cv2.waitKey(0)
cv2.destroyAllWindows()

2. 读取CMYK像素值并转为RGB显示

如果需要正常查看图像内容，建议将CMYK转为OpenCV支持的BGR格式，同时保留CMYK像素值的读取：

import numpy as np
from PIL import Image
import cv2

with Image.open("image.png") as im:
    im_cmyk = im.convert("CMYK")
    img_cmyk = np.array(im_cmyk)
    
    # 读取指定坐标的CMYK像素值，示例为(100,100)位置
    y, x = 100, 100
    c, m, y_val, k = img_cmyk[y, x]
    print(f"CMYK像素值：C={c}, M={m}, Y={y_val}, K={k}")
    
    # 将CMYK转为RGB，再转换为OpenCV默认的BGR格式
    im_rgb = im_cmyk.convert("RGB")
    img_bgr = cv2.cvtColor(np.array(im_rgb), cv2.COLOR_RGB2BGR)
    
    cv2.imshow('CMYK Converted to BGR', img_bgr)
    cv2.waitKey(0)
    cv2.destroyAllWindows()

注意事项 #

• 直接用cv2.imshow显示4通道CMYK数组会触发崩溃，必须先处理为3通道或转换为OpenCV支持的色彩空间。
• 读取CMYK像素值无需额外转换，直接从img_cmyk数组中通过坐标索引即可获取对应C、M、Y、K四个通道的值。

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