基于双分支注意力特征编码的条件扩散弱光图像增强方法及系统

本发明公开了一种基于双分支注意力特征编码的条件扩散弱光图像增强方法及系统，该方法通过构建基于双分支注意力的弱光图像特征编码模型和条件扩散模型，弱光图像特征编码模型包括梯度恢复模块和亮度恢复模块，采用通道注意力机制和空间注意力机制，将弱光图像的特征转移到梯度恢复模块和亮度恢复模块，对弱光图像的特征图和梯度图进行恢复；将梯度图特征和亮度图特征作为先验信息指导扩散模型的训练，因此增强后的正常光图像具有更好的质量以及清晰度。