基于注意力机制的高动态范围图像色调映射方法及装置

本发明涉及一种基于注意力机制的高动态范围图像色调映射方法及装置。该方法包括：S1、确定第一亮度通道。第一亮度通道为高动态范围图像的亮度通道。S2、基于通道注意力机制自适应调整第一亮度通道的全局亮度，以得到经过全局亮度映射的第一亮度特征图。S3、基于自注意力机制增强第一亮度特征图的局部细节，以得到第二亮度特征图。S4、根据第一亮度通道和第二亮度特征图得到第二亮度通道。第二亮度通道为低动态范围图像的亮度通道。本发明通过基于通道注意力实现自适应的亮度修饰，能够提高亮度映射的灵活性，基于自注意力机制增强色调映射中的细节保留，能够有效整合全局亮度映射和局部细节增强，得到更加优秀的色调映射结果。