Python(Pillow)调整YCbCr图像Y值后图像失色问题求助

问题 #

我用自定义代码实现RGB与YCbCr互转，不调整Y值时转换正常：

def rgb2ycbcr(im):
    xform = np.array([[.299, .587, .114], [-.1687, -.3313, .5], [.5, -.4187, -.0813]])
    ycbcr = im.dot(xform.T)
    ycbcr[:,:,[1,2]] += 128
    return np.uint8(ycbcr)
def ycbcr2rgb(im):
    xform = np.array([[1, 0, 1.402], [1, -0.34414, -.71414], [1, 1.772, 0]])
    rgb = im.astype(np.float32)
    rgb[:,:,[1,2]] -= 128
    rgb = rgb.dot(xform.T)
    np.putmask(rgb, rgb > 255, 255)
    np.putmask(rgb, rgb < 0, 0)
    return np.uint8(rgb)

尝试增加Y值提升亮度时，处理后的图像出现色彩异常：

fn = 'parrot.jpg'
im = Image.open(fn)
yuv_img_arr = rgb2ycbcr(np.array(im))
yuv_img_arr[:,:, 0] += 100
rgb_img = Image.fromarray(ycbcr2rgb(yuv_img_arr))
rgb_img.save(fn+'converted.jpg')

但相同逻辑的Matlab代码可正常实现亮度提升：

RGB=imread('parrot.jpg');
YCBCR=rgb2ycbcr(RGB);
Y=YCBCR(:,:,1);
Y=Y+100;
YCBCR(:,:,1)=Y;
RGB2 = ycbcr2rgb(YCBCR);

请问问题出在哪？如何解决？

错误原因 #

核心差异在于Python/numpy与Matlab对uint8类型数值运算的处理逻辑不同：

• Python中numpy的uint8数组执行加法时，会触发循环溢出（模256运算）：比如原Y值为200，加100后得到300，转成uint8会变成300-256=44，Y值不升反降，最终导致色彩异常。
• Matlab中对uint8数组进行数值运算时，会自动将类型提升为double，赋值回uint8数组时会自动饱和截断：超过255的数值会被设为255，低于0的设为0，不会出现循环溢出。

另外，你的rgb2ycbcr函数直接返回uint8类型的YCbCr数组，进一步放大了溢出问题——因为Y通道的合法范围通常是16-235（对应RGB的0-255），直接加100很容易超出uint8的0-255范围。

解决方法 #

修改处理流程，在调整Y值前先转换为高精度类型，调整后手动做饱和截断，避免溢出：

方案1：修改亮度调整代码 #

在调整Y值前将数组转为uint16或float32，调整后限制Y值在0-255之间，再转回uint8：

fn = 'parrot.jpg'
im = Image.open(fn)
yuv_img_arr = rgb2ycbcr(np.array(im))

# 转换为uint16避免溢出
yuv_img_arr = yuv_img_arr.astype(np.uint16)
yuv_img_arr[:,:, 0] += 100
# 饱和截断到0-255
yuv_img_arr[:,:, 0] = np.clip(yuv_img_arr[:,:, 0], 0, 255)
# 转回uint8
yuv_img_arr = yuv_img_arr.astype(np.uint8)

rgb_img = Image.fromarray(ycbcr2rgb(yuv_img_arr))
rgb_img.save(fn+'converted.jpg')

方案2：优化YCbCr转换函数 #

让rgb2ycbcr返回浮点类型的结果，调整Y值后再做饱和处理，这样可以避免中间步骤的溢出：

def rgb2ycbcr(im):
    xform = np.array([[.299, .587, .114], [-.1687, -.3313, .5], [.5, -.4187, -.0813]])
    # 先转float32运算，避免溢出
    ycbcr = im.astype(np.float32).dot(xform.T)
    ycbcr[:,:,[1,2]] += 128
    return ycbcr  # 返回float类型，不转uint8

def ycbcr2rgb(im):
    xform = np.array([[1, 0, 1.402], [1, -0.34414, -.71414], [1, 1.772, 0]])
    rgb = im.astype(np.float32)
    rgb[:,:,[1,2]] -= 128
    rgb = rgb.dot(xform.T)
    np.putmask(rgb, rgb > 255, 255)
    np.putmask(rgb, rgb < 0, 0)
    return np.uint8(rgb)

# 亮度调整代码
fn = 'parrot.jpg'
im = Image.open(fn)
yuv_img_arr = rgb2ycbcr(np.array(im))
yuv_img_arr[:,:, 0] += 100
# 对Y通道做饱和截断
yuv_img_arr[:,:, 0] = np.clip(yuv_img_arr[:,:, 0], 0, 255)
rgb_img = Image.fromarray(ycbcr2rgb(yuv_img_arr))
rgb_img.save(fn+'converted.jpg')

这样处理后，Y值会被正确提升且不会溢出，最终图像只会提升亮度而不会出现色彩异常。

内容的提问来源于stack exchange，提问作者Yusuf Kürşat Tuncel