Python脚本中EXR法线通道（Normal Pass）转RGB映射出现伪影与失真的问题求助

问题描述 #

我正在开发一个Python脚本，用于从EXR通道数据生成图像，服务于一个需要精确法线映射的大型项目。脚本读取EXR文件的X、Y、Z通道，并将它们映射到输出图像的RGB通道，但输出图像存在明显的伪影和失真，导致法线映射和预期结果不符。

最小可复现示例 #

下面是一个能复现问题的最小脚本，需要在Blender的Python环境中运行。它使用OpenEXR读取16位EXR文件，处理法线通道，然后用Pillow保存为PNG图像。

import bpy
import OpenEXR
import Imath
import numpy as np
from PIL import Image
import os

def process_normal_pass(exr_path, output_path):
    # Open EXR file
    exr = OpenEXR.InputFile(exr_path)
    header = exr.header()
    channels = header['channels']
    
    # Get image dimensions
    dw = header['dataWindow']
    width = dw.max.x - dw.min.x + 1
    height = dw.max.y - dw.min.y + 1
    
    # Process Normal Pass channels (assuming 16-bit)
    channel_data = []
    for channel in ['X', 'Y', 'Z']:
        full_channel_name = f'ViewLayer.Normal.{channel}'
        pixel_type = Imath.PixelType(Imath.PixelType.HALF)  # 16-bit
        channel_str = exr.channel(full_channel_name, pixel_type)
        channel_array = np.frombuffer(channel_str, dtype=np.float16).reshape(height, width)
        channel_data.append(channel_array)
    
    # Convert normal vectors to RGB colors
    x, y, z = channel_data
    
    # Normalize the vectors
    magnitude = np.sqrt(x*x + y*y + z*z)
    magnitude = np.where(magnitude > 0, magnitude, 1)
    x = x / magnitude
    y = y / magnitude
    z = z / magnitude
    
    # Map normal components to RGB (world space)
    r = np.abs(x)  # X component to Red
    g = np.zeros_like(y)  # Suppress green channel
    b = np.abs(z)  # Z component to Blue
    
    # Create the RGB representation
    layer_data = np.dstack([r, g, b])
    
    # Scale to 16-bit range and convert
    layer_data = layer_data * 65535.0
    layer_data = np.clip(layer_data, 0, 65535.0)
    layer_data = layer_data.astype(np.uint16)
    
    # Save as 16-bit PNG
    img = Image.fromarray(layer_data, mode='RGB')
    img.save(output_path, format='PNG', optimize=False)
    print(f"Normal pass image saved to {output_path}")

# Example usage
exr_file = bpy.path.abspath('//normal_pass.exr')
output_png = bpy.path.abspath('//normal_pass_output.png')

process_normal_pass(exr_file, output_png)

依赖项 #

• OpenEXR
• Imath
• numpy
• Pillow
• Blender的bpy模块（可选，仅在Blender环境运行时需要，用于bpy.path.abspath）

运行环境 #

• Blender Python环境或自定义Python环境

运行步骤 #

1. 确保所需Python包（OpenEXR、Imath、numpy、Pillow）已安装在Blender Python环境或自定义环境中。
2. 将代码中的'//normal_pass.exr'替换为你的EXR文件实际路径。
3. 在Blender脚本编辑器或自定义编辑器中运行脚本。
4. 检查输出PNG图像是否存在伪影。

预期与实际结果 #

• 预期结果：法线向量的平滑、精确表示，无可见伪影或失真。（对应参考图像：平滑的法线映射图，无明显线条干扰）
• 实际结果：输出图像中出现密集的红、蓝色线条，破坏了法线映射的均匀性和平滑度。（对应实际输出图：带有大量线条干扰的法线图）

已尝试的修改方案 #

我尝试了两种修改来解决伪影问题，但都没有得到理想结果：

1. 保留绿色通道 #

原本代码中我将绿色通道设为0，后来改为保留Y分量：

# 原代码
r = np.abs(x)  # X component to Red
g = np.zeros_like(y)  # Suppress green channel
b = np.abs(z)  # Z component to Blue

# 修改后
r = np.abs(x)
g = np.abs(y)  # Preserving the Y component
b = np.abs(z)

结果：输出效果变得更差，完全不像预期的法线图。（对应修改后图像：严重失真的法线图）

2. 另一种映射方式 #

尝试将法线分量从[-1, 1]范围映射到[0, 1]：

# 修改代码
r = (x + 1.0) * 0.5
g = (y + 1.0) * 0.5
b = (z + 1.0) * 0.5

结果：效果依然不理想，因为法线映射应该是世界空间的，而不是相机或切线空间的。（对应修改后图像：不符合预期的法线图）

尽管做了这些尝试，输出图像还是无法准确表示法线通道。我需要帮助找出失真的根本原因，并找到可靠的解决方案，正确地将X、Y、Z通道映射到RGB而不引入伪影。

备注：内容来源于stack exchange，提问作者binatheis