此代码会读取两张图片,一张用于保留高光细节,另一张用于保留暗部细节。两张图片按指定比例进行像素融合,最终生成一张合成的HDR图片。
import cv2
import numpy as np
def hdr_fusion(highlight_img_path, shadow_img_path, output_path, alpha=0.5):
    """
    HDR融合算法,读取两张图片并根据比例进行像素融合。
    
    :param highlight_img_path: 高光细节图片路径
    :param shadow_img_path: 暗部细节图片路径
    :param output_path: 合成图片的输出路径
    :param alpha: 高光图片的权重,范围为0到1
    """
    # 读取两张图片
    highlight_img = cv2.imread(highlight_img_path)
    shadow_img = cv2.imread(shadow_img_path)
    if highlight_img is None or shadow_img is None:
        print("无法读取图片,请检查文件路径。")
        return
    
    # 确保两张图片大小一致
    if highlight_img.shape != shadow_img.shape:
        print("两张图片大小不一致,调整大小...")
        shadow_img = cv2.resize(shadow_img, (highlight_img.shape[1], highlight_img.shape[0]))
    # 像素融合公式: result = alpha * highlight_img + (1 - alpha) * shadow_img
    fused_img = cv2.addWeighted(highlight_img, alpha, shadow_img, 1 - alpha, 0)
    # 保存结果图片
    cv2.imwrite(output_path, fused_img)
    print(f"融合完成,结果已保存至:{output_path}")
if __name__ == "__main__":
    # 输入图片路径
    highlight_img_path = "./highlight.jpg"  # 高光细节图片路径
    shadow_img_path = "./shadow.jpg"       # 暗部细节图片路径
    output_path = "./hdr_result.jpg"       # 合成图片保存路径
    # 设置高光图片的权重 (0到1之间)
    alpha = 0.5
    # 调用HDR融合函数
    hdr_fusion(highlight_img_path, shadow_img_path, output_path, alpha)
效果测试(网图,学习使用,侵权删)
暗部细节:

亮部细节

融合效果:




















