在 Python 中批量修复多个通道的 **inf（无穷大）** 和 **NaN（非数字）** 值，通常用于处理图像数据（如 Numpy 数组格式的图像数据），特别是在处理 EXR 文件、深度图、法线图等高动态范围图像时。 我们可以使用 **NumPy 库** 来高效地批量处理图像数据中的这些异常值。 --- ### ✅ 示例代码：使用 NumPy 批量修复 inf/NaN 值 ```python import numpy as np def replace_inf_nan(image, replacement=0.0): """ 替换图像数据中的 inf 和 NaN 值为指定值（默认为 0） 参数: image (np.ndarray): 输入图像数据，形状为 (height, width, channels) replacement (float): 用来替换 inf/NaN 的值 返回: np.ndarray: 修复后的图像数据 """ # 创建一个与图像形状相同的掩码，标记出 inf 或 NaN 的像素 mask = np.isinf(image) | np.isnan(image) # 使用指定值替换所有 inf/NaN 像素 image[mask] = replacement return image # 示例：生成一个包含 inf 和 NaN 的图像 image_data = np.array([ [[1.0, 2.0, np.inf], [3.0, np.nan, 5.0]], [[np.inf, 6.0, 7.0], [8.0, np.nan, np.inf]] ], dtype=np.float32) # 打印原始图像 print("原始图像:") print(image_data) # 替换 inf/NaN 值 cleaned_image = replace_inf_nan(image_data, replacement=0.0) # 打印修复后的图像 print("\n修复后的图像:") print(cleaned_image) ``` --- ### ✅ 代码解释： - `np.isnan(image)`：检测图像中所有的 NaN 像素。 - `np.isinf(image)`：检测图像中所有的 inf（无穷大）像素。 - `mask = np.isinf(image) | np.isnan(image)`：合并两个布尔掩码，标记出所有异常像素。 - `image[mask] = replacement`：将所有异常像素替换为指定值（默认为 0）。 --- ### ✅ 扩展功能：读取 EXR 并修复 inf/NaN 值（使用 OpenEXR 和 NumPy） ```python import OpenEXR import Imath import numpy as np import cv2 # 可选，用于保存修复后的图像 def read_exr_channel(exr_file, channel_name, shape): """读取 EXR 文件的指定通道""" pt = Imath.PixelType(Imath.PixelType.FLOAT) raw_bytes = exr_file.channel(channel_name, pt) channel = np.frombuffer(raw_bytes, dtype=np.float32) return channel.reshape(shape) def write_exr_channel(exr_file, channel_name, data): """写入修复后的通道回 EXR 文件""" pt = Imath.PixelType(Imath.PixelType.FLOAT) exr_file.writeChannel(channel_name, data.tobytes(), pt) def repair_exr_channels(exr_path, output_path, height, width, channels_to_repair): """ 修复 EXR 文件中指定通道的 inf/NaN 像素，并保存为新的 EXR 文件 参数: exr_path (str): 输入 EXR 文件路径 output_path (str): 输出 EXR 文件路径 height (int): 图像高度 width (int): 图像宽度 channels_to_repair (list): 要修复的通道列表，如 ['R', 'G', 'B', 'A', 'Z'] """ exr_file = OpenEXR.InputFile(exr_path) header = exr_file.header() out_file = OpenEXR.OutputFile(output_path, header) for channel in channels_to_repair: print(f"正在修复通道: {channel}") data = read_exr_channel(exr_file, channel, (height, width)) data = replace_inf_nan(data, replacement=0.0) write_exr_channel(out_file, channel, data) out_file.close() exr_file.close() ``` --- ##