• Stars
    star
    179
  • Rank 214,039 (Top 5 %)
  • Language
    Python
  • License
    Apache License 2.0
  • Created over 2 years ago
  • Updated over 2 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

A Flexible and Unified Image Restoration Framework (PyTorch), including state-of-the-art image restoration model. Such as NAFNet, Restormer, MPRNet, MIMO-UNet, SCUNet, SwinIR, HINet, etc. ⭐⭐⭐⭐⭐⭐

PWC PWC PWC PWC PWC PWC PWC PWC PWC PWC PWC PWC PWC PWC PWC PWC PWC

A Flexible and Unified Image Restoration Framework.

GitHub stars GitHub forks download visitors

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

Installation

See INSTALL.md for the installation of dependencies required to run FUIR.

News

  • 2022-5-17 Update the State-of-the-Art models, such as NAFNet, Restormer, MPRNet, HINet, MIMO-UNet

Model Zoo

NAFNet

网络模块结构如下所示:

模块设计如下:

两大经典图像恢复实验结果如下:

  1. Image Denoising (在SIDD数据集实现SOTA):

  1. Image Deblurring (在GoPro数据集实现SOTA):

模型训练配置

模型训练代码

Restormer

网络模块结构如下所示:

三大图像恢复实验结果对比图:

  1. Image Deraining(目前保持SOTA):

  1. Image Deblurring:

  1. Image Denoising:

模型训练配置

模型训练代码

HINet

网络模块结构如下所示:

三大图像恢复实验结果:

  1. Image Denoising:

  1. Image Deblurring:

  1. Image Deraining:

模型训练配置

模型训练代码

MPRNet

网络模块结构如下所示:

三大图像恢复实验结果:

  1. Image Deraining:
  2. Image Deblurring:
  3. Image Denoising:

模型训练配置

模型训练代码

MIMONet

网络模块结构如下所示:

图像去模糊实验结果与可视化图:

MIRNet

网络模块结构如下所示:

SCUNet

网络模块结构如下所示:

部分实验结果如下:

Experiments

Image Restoration Tasks

Task Dataset Train/Test Instructions
Image Deblurring GoPro link
Image Denoising SIDD link
Image Deblurring with JPEG artifacts REDS link
Image Derain Rain13K link
Stereo Image Super-Resolution Flickr1024+Middlebury link

Citations

If you find this project is useful, please give me a star and fork!

Contact

If you have any questions, please contact me. [email protected]


Acknowledgment

This implementation based on BasicSR and NAFNet.

statistics

visitors

More Repositories

1

awesome_lightweight_networks

The implementation of various lightweight networks by using PyTorch. such as:MobileNetV2,MobileNeXt,GhostNet,ParNet,MobileViT、AdderNet,ShuffleNetV1-V2,LCNet,ConvNeXt,etc. ⭐⭐⭐⭐⭐
Python
672
star
2

CVPR_2021_Papers

CVPR2021最新论文汇总,主要包括:Transformer, NAS,模型压缩,模型评估,图像分类,检测,分割,跟踪,GAN,超分辨率,图像恢复,去雨,去雾,去模糊,去噪,重建等等
549
star
3

EPSANet

Python
213
star
4

Awesome-AI-algorithm

人工智能算法方面的综合资料合集:包括求职面试、机器学习、深度学习、强化学习等方面的资料和代码
203
star
5

awesome-papers

机器学习,深度学习,自然语言处理,计算机视觉方面的顶级期刊会议论文集
111
star
6

awesome-machine-learning

A curated list of awesome machine Learning tutorials,courses and communities.
32
star
7

Awesome_vision_transformer

Implementation of vision transformer. ⭐⭐⭐
Python
18
star
8

Cryptology

密码学中经典算法以及RSA、EIGamal数字签名技术
C++
16
star
9

deep-learning

A curated list of awesome Deep Learning tutorials, projects and communities.
12
star
10

Awesome-NLP-Resources

自然语言处理方面资料集
10
star
11

Data_Structures_and_Algorithms

数据结构:栈、队列、串、单链表、霍夫曼编码等
C++
8
star
12

Garden-Iris

鸢尾花的分类
Jupyter Notebook
8
star
13

Image-Classification

Implement a few key architectures for image classification by using neural network
Jupyter Notebook
7
star
14

Kaggle_Bicycle_Example

Kaggle自行车租赁预测比赛
Jupyter Notebook
6
star
15

Titanic

Kaggle竞赛:预测泰坦尼克号中乘客的获救概率以及存活情况
R
6
star
16

ImageProcessing

利用VS2017平台实现图像处理的一些基本算法。图形平滑、图像锐化、图像增强、傅里叶变换、行程编码
C++
6
star
17

Router

对公共场合中商店路由器的使用情况进行用户数据分析
Java
5
star
18

knowledge_distillation

一款即插即用的知识蒸馏工具包
4
star
19

Deep_Learning_tensorflow

简单的tensorflow框架知识,softmax分类器,手写数字识别,线性回归,验证码生成与识别
Jupyter Notebook
3
star
20

LeetCode-Algorithm

剑指Offer & LeetCode Problems' Solutions: A Record of My Problem Solving Journey(Python,C++))
3
star
21

Python3-Web-Spider

Python爬虫实战案例
Jupyter Notebook
2
star