首页
/ 推荐使用:DehazeFormer —— 视觉Transformer在单张图片去雾中的创新应用

推荐使用:DehazeFormer —— 视觉Transformer在单张图片去雾中的创新应用

2024-05-21 08:02:25作者:钟日瑜

1、项目介绍

在图像处理领域,去除雾霾(图像去雾)是一项基础而重要的任务。随着深度学习的发展,尤其是卷积神经网络的应用,图像去雾技术已经取得了显著进步。然而,最近在高层视觉任务中取得突破的Transformer模型,在图像去雾领域的应用却并未引起广泛关注。DehazeFormer项目应运而生,它以Swin Transformer为基础,通过一系列改进设计,如定制化的归一化层、激活函数以及空间信息聚合策略,为图像去雾带来了全新的解决方案。

2、项目技术分析

DehazeFormer对传统的Transformer架构进行了优化,针对图像去雾任务的特点,提出了一系列创新点。其中包括修改的归一化层和激活函数,旨在更好地适应低层次视觉任务的需求。此外,该项目还采用了一种独特的空间信息聚合方案,有效提升了模型在处理复杂环境下的表现力。

3、项目及技术应用场景

DehazeFormer不仅适用于常规的室内图像去雾场景,还特别针对复杂的户外远程遥感图像进行了优化,提供了一个大型的实时遥感图像去雾数据集RS-Haze-RGB。这使得该技术能够在处理高度非均匀雾霾的情况下,展现出强大的性能。无论是用于增强现实、自动驾驶还是无人机监控等需要清晰视图的场景,DehazeFormer都能提供高效且高质量的去雾效果。

4、项目特点

  • 高效性:与现有的FFA-Net相比,即使小规模模型也能实现更优性能,参数量仅为四分之一,计算成本降低五分之一。
  • 突破性成果:大型模型是首个在SOTS室内测试集中PSNR超过40dB的方法,大幅超越现有最佳结果。
  • 易于使用:项目提供了详细的训练和评估脚本,以及预训练模型和数据集的下载链接,方便研究人员快速上手。
  • 广泛兼容:支持PyTorch 1.10.2版本,并已在CUDA 11.3和cuDNN 8.2.0环境下测试通过。

如果你正在寻找一个性能强大、效果出众且易于实施的图像去雾工具,那么DehazeFormer绝对值得尝试。立即加入社区,体验Transformer在低级视觉任务中的魅力吧!

参考文献:
@article{song2023vision,
  title={Vision Transformers for Single Image Dehazing},
  author={Song, Yuda and He, Zhuqing and Qian, Hui and Du, Xin},
  journal={IEEE Transactions on Image Processing},
  year={2023},
  volume={32},
  pages={1927-1941}
}

查看项目仓库 | 在线演示 | 论文链接 | GoogleDrive数据 | 百度网盘数据

登录后查看全文
热门项目推荐
相关项目推荐