融合噪声自适应扩散与光照感知的弱光图像增强方法

本发明公开了融合噪声自适应扩散与光照感知的弱光图像增强方法，首先设计基于Retinex理论的反射‑光照分解模块RID，提取输入弱光图像的反射率分量和光照分量；然后设计自适应扩散反射去噪模块ADRD，对反射率分量进行去噪增强，输出增强后的反射率图像；引入噪声估计网络，根据反射率图中区域的噪声强度动态调节扩散模型的去噪幅度；设计光照感知增强模块IEM，融合注意力机制与局部增强块，增强光照图像，输出增强后的光照分量；搭建完整的神经网络框架，对所提模块联合训练，最后测试保存的神经网络的模型，输出增强图像。本发明解决了现有技术中存在的增强后的图像存在细节丢失、噪声残留以及光照不均的问题。