来源:期刊VIP网 时间:
作者:武霁;丁冰;丁洁;
单位:太原理工大学电气与动力工程学院;北京理工大学集成电路与电子学院;
摘要:现有的弱光图像增强方法大多存在色彩失真,去噪效果不佳,严重依赖成对数据集进行训练等问题。针对上述问题,提出一种基于生成对抗网络的弱光图像增强方法。该模型分为生成器和判别器两部分,生成器部分使用添加EMA注意力的改进UNet网络进行图像增强,判别器部分采用包括颜色判别器、灰度判别器和多尺度判别器的多分支判别器进行融合判别图像的真实性。实验结果表明,文中方法在公开数据集上取得了优异的效果,在PSNR、SSIM、NIQE、BRISQUE等多项评价指标上有了显著提升,进一步证明了所提方法的有效性和鲁棒性。
关键词:弱光图像;;无监督学习;;生成器;;判别器;;注意力机制;;图像增强
基金资助:国家重点研发计划(2022YFB3204600);; 北京理工大学青年教师学术启动计划