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