兵器装备工程学报

文章详情

稿件标题: AL-GAN:一种融合注意力机制的轻量级GAN水下图像增强模型
稿件作者: 冯建新1,2,韩亚军1,2,潘成胜1,2,孙传林1,2,蔡远航1,2
DOI: 10.11809/bqzbgcxb2024.02.017
科学编辑: 杨杰 博士(武汉理工大学 教授)
栏目名称: 基础理论与应用研究
关键词: 水下图像增强;注意力机制;轻量级;生成对抗网络
文章摘要: 针对水下图像存在对比度低、颜色失真和现有网络模型推理速度慢等问题,提出一种融合注意力机制的轻量级GAN水下图像增强模型。该模型使用PatchGAN作为判别网络,生成网络在FUnIEGAN模型基础上,使用轻量化模型MobileNet替换原UNet特征提取网络中参数量极大的VGG16模型作为特征提取模块,提取水下退化图像特征,使网络模型参数量减少,加快了网络模型的推理速度。进一步在特征提取模块引入通道和空间注意力机制,增强了网络特征提取能力,达到了增强图像细节的目的。在EUVP数据集上进行实验,结果表明:该方法在处理真实水下图像时有很好的效果。与几种现有方法相比,本文中所提方法能够更好地提升对比度,修正色偏,减少图像细节信息损失,在主客观指标上都优于现有方法。
稿件基金: 冯建新(1975—),女,博士,副教授,Email:fengjianxin863@163.com。
引用本文格式: 冯建新,韩亚军,潘成胜,等.ALGAN:一种融合注意力机制的轻量级GAN水下图像增强模型[J].兵器装备工程学报,2024,45(2):135-143.
FENG Jianxin, HAN Yajun, PAN Chengsheng, et al.ALGAN: A lightweight GAN underwater image enhancement incorporating attention mechanisms[J].Journal of Ordnance Equipment Engineering,2024,45(2):135-143.
刊期名称: 2024年02期
出版时间: 2024年2月
收稿日期: 2023年3月15日
修回日期: 2023年4月6日
录用日期: 2023年5月6日
上线时间: 2024年2月28日
浏览次数: 1500
下载次数: 74
免费阅读PDF 在线阅读 下载本期目录 下载本期封面