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

  • Post author:
  • Post category:其他



【导读】

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


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

20bb8ba67476a18c068b64fd47f8c860.png


项目介绍

当前,对于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


数据集下载与配置:

fd8f9a0bf303528367f677f8f76ce299.png

  1. 图像去噪数据集 SIDD

  2. 图像去模糊数据集 GoPro

  3. 图像去模糊数据集REDS

  4. 图像去雨数据集 Rain13K


Model Zoo

c79ab155d39e3dc9eea234df605add56.png
3a3c4d396ea6bb1bc7b8424cc7c28875.png
462056d64f81f7359ecd48b44c18601c.png
098c03dceaee57aaefe753e88722a7ad.png

更多详细的模型结构、模型训练与测试信息见:

https://github.com/murufeng/FUIR

推荐阅读


欢迎大家加入DLer-


计算机视觉技术交流


群!

大家好,群里会第一时间发布

计算机视觉方向的前沿论文解读和交流分享

,主要方向有:

图像分类、Transformer、目标检测、目标跟踪、点云与语义分割、GAN、超分辨率、人脸检测与识别、动作行为与时空运动、模型压缩和量化剪枝、迁移学习、人体姿态估计


等内容。


进群请备注:


研究方向+学校/公司+昵称


(如图像分类


+上交+小明)

8f63d5cd302dcb51473453fcacc8cdd7.png

👆

长按识别,邀请您进群!



版权声明:本文为weixin_47196664原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。