【重磅开源】一文汇总顶会 SOTA 图像恢复算法,包括图像去噪、去雨、去模糊等等...

【导读】今天给大家推荐一个很棒的开源项目,该项目主要汇总了一系列Low-Level-vision领域当前主流的SOTA算法网络结构,主要包括图像去噪,图像去雨、图像去模糊等。该项目详细介绍了各大主流Low-Level-Vison数据集(如SIDD、GoPro、REDS、Rain13K等)的下载链接和训练配置教程,每个主流的算法网络结构都配备训练和测试命令方法。小伙伴们,快去Star关注吧!

项目地址:https://github.com/murufeng/FUIR

【重磅开源】一文汇总顶会 SOTA 图像恢复算法,包括图像去噪、去雨、去模糊等等..._第1张图片

项目介绍

当前,对于Low-level-Vision中的图像恢复任务(Image Super-resolution, Image Denoising, Image Deblurring等) 已经出现了很多优秀实用的工具包。但是,对于去年出现的一系列Transformer网络架构还没有一个统一的项目将它集成起来,本项目主要在具体图像恢复任务的数据处理和网络训练配置上将当前主流的CNN网络和基于Transformer的网络架构集成了起来。本项目将作为图像恢复任务的一个灵活统一的工具。利用本项目可以快速实现图像去噪,图像/视频去模糊,图像去雨等一系列经典任务的训练。本项目提供在GOPRO、SIDD、REDS、Rain13K数据集上的数据预处理以及网络架构训练教程,后续我们将持续更新。希望本项目既能让图像处理初学者快速入门,又能服务科研和工业社区。

项目安装命令:

  1. Star and Clone our repository

git clone https://github.com/murufeng/FUIR.git 
cd FUIR
  1. Make conda environment

conda create -n pytorch1.9.0_FUIR python=3.7 conda activate pytorch1.9.0_FUIR
  1. Install dependencies

conda install pytorch=1.9.0 torchvision cudatoolkit=11.2 -c pytorch pip install matplotlib scikit-learn scikit-image opencv-python yacs joblib natsort h5py tqdm pip install einops gdown addict future lmdb numpy pyyaml requests scipy tb-nightly yapf lpip
  1. Build

python setup.py develop --no_cuda_ext

数据集下载与配置:

【重磅开源】一文汇总顶会 SOTA 图像恢复算法,包括图像去噪、去雨、去模糊等等..._第2张图片
  1. 图像去噪数据集 SIDD

  2. 图像去模糊数据集 GoPro

  3. 图像去模糊数据集REDS

  4. 图像去雨数据集 Rain13K

Model Zoo

【重磅开源】一文汇总顶会 SOTA 图像恢复算法,包括图像去噪、去雨、去模糊等等..._第3张图片 【重磅开源】一文汇总顶会 SOTA 图像恢复算法,包括图像去噪、去雨、去模糊等等..._第4张图片 【重磅开源】一文汇总顶会 SOTA 图像恢复算法,包括图像去噪、去雨、去模糊等等..._第5张图片 【重磅开源】一文汇总顶会 SOTA 图像恢复算法,包括图像去噪、去雨、去模糊等等..._第6张图片

更多详细的模型结构、模型训练与测试信息见:https://github.com/murufeng/FUIR

推荐阅读

  • 西电IEEE Fellow团队出品!最新《Transformer视觉表征学习全面综述》

  • 如何做好科研?这份《科研阅读、写作与报告》PPT,手把手教你做科研

  • 最新 2022「深度学习视觉注意力 」研究概述,包括50种注意力机制和方法!

  • 【重磅】斯坦福李飞飞《注意力与Transformer》总结,84页ppt开放下载!

  • 分层级联Transformer!苏黎世联邦提出TransCNN: 显著降低了计算/空间复杂度!

  • 清华姚班教师劝退文:读博,你真的想好了吗?

  • 2021李宏毅老师最新40节机器学习课程!附课件+视频资料

欢迎大家加入DLer-计算机视觉技术交流群!

大家好,群里会第一时间发布计算机视觉方向的前沿论文解读和交流分享,主要方向有:图像分类、Transformer、目标检测、目标跟踪、点云与语义分割、GAN、超分辨率、人脸检测与识别、动作行为与时空运动、模型压缩和量化剪枝、迁移学习、人体姿态估计等内容。

进群请备注:研究方向+学校/公司+昵称(如图像分类+上交+小明)

【重磅开源】一文汇总顶会 SOTA 图像恢复算法,包括图像去噪、去雨、去模糊等等..._第7张图片

 长按识别,邀请您进群!

你可能感兴趣的:(机器学习,人工智能,深度学习,计算机视觉,python)