DehazeNet: An End-to-End System for Single ImageHaze Removal(图像去雾2016)

摘要:

在这篇文章中,提出了一个可训练的端到端系统称为DehazeNet,用于介质传输估计。DehazeNet将一幅有雾的图像作为输入,输出其介质透射图,然后通过大气散射模型恢复出一幅无雾的图像。DehazeNet采用基于卷积神经网络(CNN)的深度架构,其层是专门设计来体现图像去雾中的既定假设/先验。具体来说,Maxout单元层用于特征提取,可以生成几乎所有的与雾向光的特征。我们还在去雾网中提出了一种新的非线性激活函数,称为双边校正线性单元(BReLU ),它能够提高恢复的无雾图像的质量。在提出的去雾网的组件和现有方法中使用的组件之间建立联系。

本文贡献:

(1)DehazeNet是一个端到端的系统。它直接学习和估计模糊图像块与其介质传输之间的映射关系。这是通过其深层架构的特殊设计来实现的,以体现既定的图像去雾原则。

(2)我们在DehazeNet中提出了一种新的非线性激活函数,称为双边校正线性单元1(BReLU)。BReLU扩展了校正线性单元(ReLU ),并证明了它在获得精确图像恢复中的重要性。在技术上,BReLU使用双边约束来减少搜索空间和提高收敛性。

(3)我们建立了去雾网的组件与现有去雾方法中使用的那些假设/先验之间的联系,并解释了去雾网通过端到端自动学习所有这些组件来改进这些方法

related work

大气散射模型

I(x)是观测到的朦胧图像,J(x)是要复原的真实场景,t(x)是介质透射,α是全球大气光

 d(x)是场景点到摄像机的距离,β是大气的散射系数

 d(x)趋于无穷大时,t(x)趋近于零

全局大气光的估计

相关工作

暗通道:

暗通道先验是基于对室外无霾图像的广泛观察。在大多数无霾色块中,至少一个颜色通道具有一些亮度值非常低甚至接近于零的像素。暗通道[9]定义为局部区域中所有像素颜色的最小值:

DehazeNet是一种专门用于单幅图像的深度学习框架,其设计目标是通过端到端的学习方式有效地估计和图像中的霾效应。该方法采用了卷积神经网络(CNN)结构,并结合了物理成像模型,以实现高质量的效果[^1]。 在DehazeNet中,输入图像被送入一个多层次的网络中,该网络包括多个卷积层、池化层以及非线性激活函数。网络的核心思想是直接从有图像中学习传输图(transmission map),这是基于大气散射模型的关键参数之一。通过对大量带有合成霾效果的数据集进行训练,DehazeNet能够准确地预测出每张输入图像对应的传输图,并进一步利用这个传输图来恢复清晰的无图像[^1]。 此外,DehazeNet还引入了一些创新性的技术细节,例如使用多尺度特征融合策略来增强对不同尺度霾特征的捕捉能力,以及采用后处理步骤来优化最终输出的视觉质量。这些改进使得DehazeNet相较于传统的方法,在保持边缘细节的同时还能提供更加自然的结果[^1]。 以下是一个简化的DehazeNet架构示意图,用以展示其基本组成部分: ```python # 示例代码仅用于说明目的,实际实现会更复杂 import torch.nn as nn class DehazeNet(nn.Module): def __init__(self): super(DehazeNet, self).__init__() # 定义卷积层 self.conv1 = nn.Conv2d(in_channels=3, out_channels=64, kernel_size=3, padding=1) self.conv2 = nn.Conv2d(in_channels=64, out_channels=64, kernel_size=3, padding=1) # 更多层定义... # 激活函数 self.relu = nn.ReLU() # 输出层 self.out_conv = nn.Conv2d(in_channels=64, out_channels=3, kernel_size=3, padding=1) def forward(self, x): # 前向传播过程 x = self.relu(self.conv1(x)) x = self.relu(self.conv2(x)) # ... 其他层操作 ... transmission_map = self.out_conv(x) return transmission_map ``` 请注意,上述代码仅为简化版的DehazeNet模型示意,真实的DehazeNet实现将包含更多的细节和技术考量,如特定的损失函数设计、数据增强策略等。
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值