3 个月前

基于反向投影金字塔网络的多样化雾霾场景单图像去雾

基于反向投影金字塔网络的多样化雾霾场景单图像去雾

摘要

单张雾霾图像去雾任务,尤其是在训练数据集较小的情况下,极具挑战性。为此,我们提出了一种新型生成对抗网络架构——反投影金字塔网络(Back Projected Pyramid Network, BPPNet),该方法在多种复杂雾霾条件下均表现出色,包括浓雾和非均匀雾。BPPNet架构通过迭代的UNet块保留了空间上下文信息,同时引入一种新颖的金字塔卷积块,以捕捉多尺度结构特征。这些模块共同构成生成器,并可通过反向投影机制有效进行学习。实验表明,我们的网络仅需20对雾霾与清晰图像即可实现有效训练,且不会出现过拟合现象。在NTIRE 2018均匀雾霾数据集(涵盖室内与室外图像)、NTIRE 2019浓雾数据集以及NTIRE 2020非均匀雾霾数据集上,我们的方法均取得了当前最优的性能表现。

代码仓库

基准测试

基准方法指标
image-dehazing-on-dense-hazeBPPNet
PSNR: 17.01
SSIM: 0.613
image-dehazing-on-i-hazeBPPNet
PSNR: 22.56
SSIM: 0.8994
image-dehazing-on-o-hazeBPPNet
PSNR: 24.27
SSIM: 0.8919

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于反向投影金字塔网络的多样化雾霾场景单图像去雾 | 论文 | HyperAI超神经