基于改进循环生成对抗网络的低照度图像增强

隋涛 ,  吴森炜 ,  贾浩 ,  万可欣 ,  杨洋

科学技术与工程 ›› 2024, Vol. 24 ›› Issue (14) :5911-5919

PDF (3375KB)
科学技术与工程 ›› 2024, Vol. 24 ›› Issue (14) :5911-5919 DOI: 10.12404/j.issn.1671-1815.2304161
自动化技术、计算机技术

基于改进循环生成对抗网络的低照度图像增强

作者信息 +

Low-light Image Enhancement Based on Improved Cycle Generative Adversarial Network

Author information +
文章历史 +

摘要

为了解决在低照度图像增强过程中配对数据集获取困难,且经过增强后的图像质量不佳的问题,通过改进循环生成对抗网络模型的方法研究了非配对低照度图像增强的实现。生成器部分采用融合了Vision Transformer结构的U-NET模型替代原始的生成器模型,来提高图像变换的周期一致性和内容保持性,并有效地处理图像研究中普遍存在的长距离空间相关性的问题。判别器部分针对图像研究的特点选择PatchGAN代替传统的判别器,提高对图像细节的判别能力。同时引入身份一致性损失函数,提高图像质量。结果表明,相较于传统方法,本文改进的模型有着更好的主观视觉效果,同时在客观评价指标也有着相应的提高,可见本文改进模型的有效性。

Abstract

In order to solve the problems of difficulty in obtaining paired data sets and poor quality of enhanced images in the process of low-light image enhancement, the implementation of unpaired low-light image enhancement was studied by improving the cycle generative adversarial network model. In the generator part, a U-NET model integrated with the Vision Transformer structure was used to replace the original generator model, so as to improve the cycle consistency and content retention of image transformation, and effectively deal with the problem of long-distance spatial correlation commonly existing in image research. In the discriminator part, PatchGAN was selected to replace the traditional discriminator according to the characteristics of image research to improve the discrimination ability of image details. At the same time, the identity consistency loss function was introduced to improve the image quality. The results show that compared with the traditional method, the improved model in this paper has better subjective visual effect, and also has a corresponding improvement in the objective evaluation index. It can be seen that the improved model in this paper is effective.

引用本文

引用格式 ▾
隋涛, 吴森炜, 贾浩, 万可欣, 杨洋. 基于改进循环生成对抗网络的低照度图像增强[J]. 科学技术与工程, 2024, 24(14): 5911-5919 DOI:10.12404/j.issn.1671-1815.2304161
SUI Tao, WU Sen-wei, JIA Hao, WAN Ke-xin, YANG Yang. Low-light Image Enhancement Based on Improved Cycle Generative Adversarial Network[J]. Science Technology and Engineering, 2024, 24(14): 5911-5919 DOI:10.12404/j.issn.1671-1815.2304161

登录浏览全文

4963

注册一个新账户 忘记密码

参考文献

基金资助

国家自然科学基金面上项目(62273214)

教育部协同育人项目(220900782082623)

山东省高校科研计划(J18KA317)

PDF (3375KB)

32

访问

0

被引

详细

导航
相关文章

/

〈 〉