此代码会读取两张图片,一张用于保留高光细节,另一张用于保留暗部细节。两张图片按指定比例进行像素融合,最终生成一张合成的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)
效果测试(网图,学习使用,侵权删)
暗部细节:
亮部细节
融合效果: