如何使用Python的OpenEXR或OpenCV为RGB格式EXR文件添加Alpha通道

你写的两段代码都存在逻辑和格式错误，无法正确生成带Alpha通道的EXR文件，以下是修正后的可运行版本，以及原代码的问题说明。

OpenEXR 实现版（兼容性最佳） #

直接操作EXR原生通道，不会出现通道顺序错乱问题，适合对EXR参数有定制需求的场景：

import OpenEXR
import Imath
import numpy as np
import os

def write_exr_alpha(input_path, output_path, alpha_threshold=100):
    # 读取输入文件
    in_exr = OpenEXR.InputFile(input_path)
    float_type = Imath.PixelType(Imath.PixelType.FLOAT)
    in_header = in_exr.header()

    # 计算图像尺寸
    data_window = in_header['dataWindow']
    img_w = data_window.max.x - data_window.min.x + 1
    img_h = data_window.max.y - data_window.min.y + 1

    # 读取RGB三通道，保留二维像素结构
    channel_data = {}
    for c in ["R", "G", "B"]:
        raw = in_exr.channel(c, float_type)
        channel_data[c] = np.frombuffer(raw, dtype=np.float32).reshape(img_h, img_w)

    # 生成Alpha通道：按原逻辑，R通道值大于阈值的位置透明度设为0，其余不透明
    alpha = np.ones((img_h, img_w), dtype=np.float32)
    alpha[channel_data["R"] > alpha_threshold] = 0

    # 输出文件
    os.makedirs(os.path.dirname(output_path), exist_ok=True)
    out_header = OpenEXR.Header(img_w, img_h)
    out_header["compression"] = in_header["compression"] # 保留原压缩格式
    out_exr = OpenEXR.OutputFile(output_path, out_header)
    out_exr.writePixels({
        "R": channel_data["R"].tobytes(),
        "G": channel_data["G"].tobytes(),
        "B": channel_data["B"].tobytes(),
        "A": alpha.tobytes()
    })
    out_exr.close()

OpenCV 实现版（代码更简洁） #

依赖OpenCV的EXR读写模块，注意处理OpenCV默认的BGR通道顺序即可：

import os
import cv2
import numpy as np

def write_exr_alpha(input_path, output_path, alpha_threshold=100):
    os.makedirs(os.path.dirname(output_path), exist_ok=True)
    # 按原始位深读取EXR
    img = cv2.imread(input_path, cv2.IMREAD_ANYCOLOR | cv2.IMREAD_ANYDEPTH)
    if img is None:
        raise RuntimeError("输入文件读取失败，请检查路径和文件格式")

    # 生成Alpha通道，cv2读入的通道顺序为B/G/R，通道0对应原B通道、通道2对应原R通道
    alpha = np.ones(img.shape[:2], dtype=img.dtype)
    # 若要按原R通道判断阈值，将img[:,:,0]改为img[:,:,2]
    alpha[img[:,:,0] > alpha_threshold] = 0

    # 合并为BGRA四通道（OpenCV原生顺序，无需调换R/B位置）
    img_rgba = cv2.merge((img[:,:,0], img[:,:,1], img[:,:,2], alpha))
    # 指定无压缩写入，避免其他软件不兼容
    cv2.imwrite(output_path, img_rgba, [cv2.IMWRITE_EXR_COMPRESSION, 0])

原代码错误点 #

• OpenEXR版本：
  • 读取通道时丢弃了G、B的原始数据，三个通道全部填充R通道值，输出为灰度图
  • Alpha通道直接传入numpy数组，没有转为writePixels要求的二进制bytes格式，写入会报错
  • 将通道数据拉平为一维列表计算Alpha，像素索引和二维图像位置不对应，Alpha区域完全错误
• OpenCV版本：
  • 合并通道时错误调换了R、B通道位置，输出图像色彩反转
  • Alpha赋值的循环逻辑错误，仅遍历了坐标元组的长度，没有遍历所有符合条件的像素点
  • 没有指定EXR写入压缩参数，默认压缩格式可能导致Nuke、PS等软件无法正常读取输出文件

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