论文标题:Contrastive Learning for Compact Single Image Dehazing
论文链接:https://arxiv.org/abs/2104.09367
代码链接:https://github.com/GlassyWu/AECR-Net
作者单位:华东师范大学 & 厦门大学 & 腾讯优图
参数与性能之间的权衡表现最佳!性能优于FFA-Net、FDU和KDDN等网络,代码即将开源!
由于严重的信息退化,单图像去雾是一个具有挑战性的不适定问题。然而,现有的基于深度学习的除雾方法仅采用清晰的图像作为正样本来指导除雾网络的训练,而未利用负信息。而且,它们中的大多数集中在通过增加深度和宽度来增强除雾网络,从而导致对计算和存储器的显着需求。在本文中,我们提出了一种基于对比学习的新颖对比正则化(CR)技术,以利用模糊图像和清晰图像的信息分别作为负样本和正样本。 CR确保在表示空间中将还原后的图像拉到更接近清晰图像,并推到远离朦胧图像的位置。此外,考虑到性能和内存存储之间的权衡,我们基于类自动编码器(AE)框架开发了一个紧凑的除雾网络。它涉及一个自适应混合操作和一个动态特征增强模块,可以分别受益于自适应地保存信息流和扩展接收域以提高网络的转换能力。我们将具有自动编码器和对比正则化功能的除雾网络称为AECR-Net。在合成和真实数据集上进行的广泛实验表明,我们的AECR-Net超越了最新技术。
内容中包含的图片若涉及版权问题,请及时与我们联系删除
评论
沙发等你来抢