如何将图像红色色相替换为蓝色？代码异常问题排查求助

问题分析 #

1. 使用HLS而非HSV导致色相判断偏差：colorsys的rgb_to_hls返回的亮度（lightness）计算方式为(max(rgb)+min(rgb))/2，会让高亮度/低亮度的红色系颜色色相计算不稳定，渐变区域的颜色容易超出设定的色相范围，无法被替换。
2. 色相范围过于严苛：仅用0<=h<=1/6或5/6<=h<=1的范围，会漏掉偏橙红、暗红等边缘红色系颜色，尤其是渐变中的过渡色。
3. 未结合饱和度过滤非目标色：低饱和度的灰色、白色等颜色可能因色相刚好落在红色范围被误替换，同时一些低饱和的红色又会被漏掉。
4. 逐像素getpixel/putpixel效率极低：处理大图片时速度慢，且容易积累浮点转换误差。

修复后的代码 #

改用HSV色彩空间进行判断和替换，同时优化颜色判断逻辑，并用更高效的像素访问方式：

from PIL import Image
import colorsys

def change_red_to_blue(image_path, output_path):
    img = Image.open(image_path).convert('RGBA')
    pixels = img.load()  # 获取像素访问对象，比getpixel/putpixel效率高很多
    width, height = img.size

    for x in range(width):
        for y in range(height):
            r, g, b, a = pixels[x, y]
            # 转换为HSV色彩空间（h: 0-1, s:0-1, v:0-1）
            h, s, v = colorsys.rgb_to_hsv(r/255.0, g/255.0, b/255.0)
            
            # 多维度判断红色系：色相覆盖红/橙红/深红范围，饱和度过滤低饱和非红色，RGB通道确保红色占优
            is_red = (h <= 0.2 or h >= 0.8) and s > 0.15 and r > max(g, b)
            if is_red:
                # 将红色色相偏移到蓝色（蓝色色相为2/3，对应240度）
                h = (h + 2/3) % 1.0
            
            # 转换回RGB并写入像素
            r_new, g_new, b_new = colorsys.hsv_to_rgb(h, s, v)
            pixels[x, y] = (int(r_new*255), int(g_new*255), int(b_new*255), a)
    
    img.save(output_path)

change_red_to_blue("C:\\color\\tes.png", "C:\\color\\tes_output.png")

关键优化点 #

• 切换到HSV色彩空间：HSV的色相（hue）对于纯色和渐变过渡色的判断更稳定，不会因亮度变化产生大幅偏移。
• 多维度红色判断：结合色相范围、饱和度和RGB通道值，精准识别红色系，避免误判和漏判渐变颜色。
• 高效像素访问：使用img.load()获取像素对象，直接修改原图像像素，比创建新图像逐像素写入快数倍。
• 放宽色相范围：将红色色相范围设为h<=0.2或h>=0.8，覆盖橙红、深红等边缘红色系，适配渐变过渡。

如果渐变区域仍有漏网的红色，可以微调以下参数：

• 降低饱和度阈值（比如s>0.1），覆盖更暗的红色
• 扩大色相范围（比如h<=0.25或h>=0.75）
• 调整RGB通道的判断条件（比如r > g + 30 and r > b +30，给红通道更高的权重）

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