如何用Python修改并保存带ICC色彩配置文件的16位RGBA图像

我有一批带ICC色彩配置文件的单通道16位深度RGBA格式PNG图像，需要移除Alpha通道并保留原ICC配置文件保存。尝试用python-opencv和PIL.Image时遇到问题：

• OpenCV可保存16位RGBA图像，但无法保留ICC配置文件
• PIL.Image可保留ICC配置文件，但似乎不支持16位RGBA图像

我是否错误使用了这两个模块？或是有其他可处理该需求的模块？

尝试代码 #

OpenCV部分 #

import cv2
import numpy as np

src_image = cv2.imdecode(np.fromfile(file_path, dtype=np.uint8), cv2.IMREAD_UNCHANGED)

# 检测位深度
if src_image.dtype == "uint8":
    depth_value = 255
if src_image.dtype == "uint16":
    depth_value = 65535

# 如果Alpha通道全白，则移除Alpha通道
b, g, r, a = cv2.split(src_image)
if np.all(a == depth_value):
    merged = cv2.merge([b,g,r])
    cv2.imwrite(out_path,merged,[cv2.IMWRITE_PNG_COMPRESSION,png_comp])
else:
    pass

PIL.Image部分 #

from PIL import Image

image = Image.open(image_path)
image.load()

r, g, b, a = image.split()
print(r.mode) # 输出"L"
image = Image.merge('BGR;16',(b,g,r)) # ValueError: wrong number of bands

image.save(out_path,bits = 'BGR;16',icc_profile=image.info.get('icc_profile')) # 只找到"BGR;16"，没有"RGB;16"

PIL.Image模式说明（翻译） #

模式
图像的模式是一个字符串，定义了图像中像素的类型和深度。每个像素使用位深度的完整范围：1位像素范围为0-1，8位像素为0-255，依此类推。当前版本支持以下标准模式：

• 1：1位像素，黑白，每个像素占1字节
• L：8位像素，黑白
• P：8位像素，通过调色板映射到其他模式
• RGB：3×8位像素，真彩色
• RGBA：4×8位像素，带透明蒙版的真彩色
• CMYK：4×8位像素，分色
• YCbCr：3×8位像素，彩色视频格式（注：指JPEG标准，而非ITU-R BT.2020标准）
• LAB：3×8位像素，Lab色彩空间
• HSV：3×8位像素，色相、饱和度、明度色彩空间
• I：32位有符号整数像素
• F：32位浮点像素

Pillow还对部分额外模式提供有限支持：

• LA：带Alpha通道的L模式
• PA：带Alpha通道的P模式
• RGBX：带填充的真彩色
• RGBa：带预乘Alpha的真彩色
• La：带预乘Alpha的L模式
• I;16：16位无符号整数像素
• I;16L：16位小端无符号整数像素
• I;16B：16位大端无符号整数像素
• I;16N：16位本机端无符号整数像素
• BGR;15：15位反转真彩色
• BGR;16：16位反转真彩色
• BGR;24：24位反转真彩色
• BGR;32：32位反转真彩色

1. 修正PIL用法 #

你遇到的问题是对PIL的16位模式理解有误：16位RGBA图像在PIL中的模式是RGBA;16（或带端序的RGBA;16L/RGBA;16B），拆分后每个通道是I;16模式而非L，之前用BGR;16合并L通道自然会报错。

正确处理代码：

from PIL import Image

image = Image.open(image_path)
icc_profile = image.info.get('icc_profile')

# 针对16位RGBA场景处理
if image.mode in ('RGBA;16', 'RGBA;16L', 'RGBA;16B'):
    r, g, b, a = image.split()
    # 检查Alpha通道是否全为16位最大值（65535）
    if all(pixel == 65535 for pixel in a.getdata()):
        # 匹配原图像的端序，合并为RGB模式
        rgb_mode = image.mode.replace('RGBA', 'RGB')
        rgb_image = Image.merge(rgb_mode, (r, g, b))
        # 保存时传入原ICC配置
        rgb_image.save(out_path, icc_profile=icc_profile, compression='zip')

2. OpenCV + PIL 组合方案 #

OpenCV本身不支持写入PNG的ICC配置文件，可先用OpenCV处理图像，再用PIL补加ICC：

import cv2
import numpy as np
from PIL import Image

# OpenCV处理图像
src_image = cv2.imdecode(np.fromfile(file_path, dtype=np.uint8), cv2.IMREAD_UNCHANGED)
depth_value = 65535 if src_image.dtype == np.uint16 else 255

b, g, r, a = cv2.split(src_image)
if np.all(a == depth_value):
    merged = cv2.merge([b, g, r])
    # 临时保存处理后的图像
    temp_path = 'temp_processed.png'
    cv2.imwrite(temp_path, merged, [cv2.IMWRITE_PNG_COMPRESSION, 9])
    
    # PIL读取临时文件并添加ICC
    with Image.open(temp_path) as processed_img:
        with Image.open(file_path) as orig_img:
            orig_icc = orig_img.info.get('icc_profile')
        processed_img.save(out_path, icc_profile=orig_icc, compression='zip')

3. 推荐使用PyVips #

PyVips对16位图像、ICC配置文件的支持更完善，操作简洁高效：

import pyvips

# 读取图像并保留ICC
image = pyvips.Image.new_from_file(file_path, access='sequential')

# 检查4通道RGBA且Alpha全白
if image.bands == 4:
    alpha_channel = image[3]
    if (alpha_channel == 65535).all():
        # 移除Alpha通道，保留前3个RGB通道
        rgb_image = image[0:3]
        # 保存时写入原ICC配置
        rgb_image.write_to_file(out_path, icc_profile=image.get('icc-profile-data'))

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