基于Transformer的弱光图像增强方法

本发明公开了基于Transformer的弱光图像增强方法，方法为：输入的弱光图像经过一个卷积层进行提取图像的局部细节信息并转换为指定大小的局部特征图；将局部特征图输入基于3D多头注意力构建三个连续的特征提取模块中以提取更深层次的全局特征；将三个全局特征通过一个多层融合模块进行融合生成最终增强的特征图；多层融合模块采用自适应加权方法融合来自不同层次的特征；增强的特征图通过一个卷积层和一个Sigmoid激活函数，得到最终增强的图像。本发明改善在低光照条件下拍摄的图像质量。