DEA-Net:基于细节增强卷积和内容引导注意力的单图像去雾

IEEE TIP 2024 | DEA-Net:基于细节增强卷积和内容引导注意力的单图像去雾 DEA-Net: Single image dehazing based on detail-enhanced convolution and content-guided attention

paper name: DEA-Net: Single image dehazing based on detail-enhanced convolution and content-guided attention

论文介绍:提出一种单图像去雾网络DEA-Net,DEA-Net是一种有监督式学习,网络在以下几点有创新之处:

1.使用多种差分卷积(中心差分卷积,角度差分卷积,水平差分卷积,垂直差分卷积)来替换普通的卷积

2.提出内容引导注意(content-guided attention),它是由空间注意力(Spatial attention)和通道注意力(Channel attention)

3.提出基于内容引导注意的混合融合方案,提出一种基于遗传算法的混合方案,通过学习空间权重来调整特征。

method

1.差分卷积

论文中大量使用了四种差分卷积来替换普通的卷积,其可以更好的提取出边缘特征。中心差分卷积的示例图如图1所示。中心差分卷积的主要操作为把输入矩阵x每个元素都减去x5,然后再与对于的卷积核进行运算。其他三种差分卷积类似,只是差分操作不一样。水平差分卷积如图2所示。

图1 中心差分卷积

图2 水平差分卷积

2.内容引导注意(CGA)

内容引导注意由空间注意力(Spatial attention)和通道注意力(Channel attention),其流程图如图3所示。

图中GMP和GAP分别代表最大池化和平均池化操作。channel shuffle操作是通过重新排列卷积层输出的通道。这样可以促进特征之间的交互,从而提高模型的表达能力。

图3 内容引导注意力

3.提出基于内容引导注意的混合融合方案

一种基于遗传算法的混合方案,通过学习空间权重来调整特征。论文中所谓的学习的空间权重就是将CGA的输出结果W。用得到的空间权重w与Low-level features和High-level features进行乘法相加得到。

图4 DEA-Net网络结构图

相关推荐
新智元2 分钟前
一句话,性能暴涨 49%!马里兰 MIT 等力作:Prompt 才是大模型终极武器
人工智能·openai
猫头虎9 分钟前
猫头虎AI分享|一款Coze、Dify类开源AI应用超级智能体Agent快速构建工具:FastbuildAI
人工智能·开源·github·aigc·ai编程·ai写作·ai-native
新智元25 分钟前
AI 版华尔街之狼!o3-mini 靠「神之押注」狂赚 9 倍,DeepSeek R1 最特立独行
人工智能·openai
天下弈星~34 分钟前
GANs生成对抗网络生成手写数字的Pytorch实现
人工智能·pytorch·深度学习·神经网络·生成对抗网络·gans
飞翔的佩奇34 分钟前
【完整源码+数据集+部署教程】食品分类与实例分割系统源码和数据集:改进yolo11-AggregatedAttention
python·yolo·计算机视觉·数据集·yolo11·食品分类与实例分割
重启的码农44 分钟前
ggml介绍 (8) 图分配器 (ggml_gallocr)
c++·人工智能·神经网络
重启的码农1 小时前
ggml介绍 (9) 后端调度器 (ggml_backend_sched)
c++·人工智能·神经网络
aneasystone本尊1 小时前
学习 Coze Studio 的智能体执行逻辑
人工智能
盏灯1 小时前
Trae SOLO 游戏 —— 🐾🐱🐾猫咪追蝌蚪🐸
人工智能·trae
lisuwen1161 小时前
AI三国杀:马斯克炮轰苹果“偏袒”OpenAI,Grok与ChatGPT的应用商店战争揭秘
人工智能·chatgpt