计算机工程

作者：刘彦红, 杨秋翔, 胡帅

单位：中北大学软件学院, 山西 太原 030051

关键词：图像去雾,特征融合,特征差异,坐标注意力,通道注意力

基金：山西省科技攻关项目(20090322004);总装预研基金(9140A17020113BQ04226)。

雾霾是大气污染物在逆温等气象条件下聚集和累积而形成的混浊物质,其可见性十分有限。图像去雾技术能够消除由雾霾导致的模糊、低对比度等问题,提高图像的清晰度和可见性,但也存在图像细节信息丢失等问题。为此,提出一种基于特征差异的多尺度特征融合去雾(FD-CA dehaze)网络。对FFA-Net的基本块结构进行改进,分别从特征差异维度、坐标维度和通道维度提取中间特征信息。提出有效坐标注意力(ECA)模块,将全局池化、最大池化与坐标位置信息相结合,用于减轻特征融合过程中的位置信息丢失问题;将通道注意力与ECA模块相结合,构建双注意力(D-CA)模型,更好地利用空间信息和通道信息,进一步提升模型在图像去雾任务中的表现。在此基础上,改进损失函数,将L1损失与感知损失相结合。在综合目标测试集(SOTS)和混合主观测试集(HSTS)中进行实验,结果表明,FD-CA dehaze网络在峰值信噪比、结构相似度2个指标上分别达到37.93 dB和0.990 5,相较于FFA-Net、GridDehazeNet等经典去雾网络,FD-CA dehaze的去雾效果得到明显提升。

来源：2024年第4期

《计算机工程》期刊编辑部

查看计算机工程杂志2024年第4期