基于扩散模型和零样本训练的高动态范围图像色调映射方法

本发明属于图像处理技术领域，具体涉及基于扩散模型和零样本训练的高动态范围图像色调映射方法。本发明包括构建色调映射网络模型，并将其集成到扩散模型的框架中，将结构信息和亮度大致分布图作为扩散模型的条件输入，利用原图的结构信息修正扩散模型反向采样的中间结果等。网络模型包括生成模型以及其控制模块；生成模型包括级联的四个尺度的编码模块、一个中间模块以及级联的四个尺度解码模块；控制模块包括和SD模型相同的四个尺度的编码模块，并在每个尺度的网络最后加入零卷积层。控制模块每个尺度的输出加至生成模型对应尺度输出的特征图上，以此控制生成模型的最终输出结果。本发明可以很好地保留原图信息，提高HDR图像色调映射效果。