Top 7 Python restoration Projects

CodeFormer

28 13,301 3.4 Python

[NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer

Project mention: Automatic1111 for Intel Arc (A380 Tested) | /r/StableDiffusion | 2023-05-29

CodeFormer

BasicSR

5 6,177 0.0 Python

Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also support StyleGAN2, DFDNet.

Project mention: About Open Source Image and Video Restoration Toolbox | news.ycombinator.com | 2024-02-27

WorkOS

workos.com sponsored

The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
SwinIR

27 4,060 0.0 Python

SwinIR: Image Restoration Using Swin Transformer (official repository)
SUPIR

3 3,359 7.3 Python

SUPIR aims at developing Practical Algorithms for Photo-Realistic Image Restoration In the Wild

Project mention: Compressing Images with Neural Networks | news.ycombinator.com | 2024-03-18

Current SOTA open source is I believe SUPIR (Example - https://replicate.com/p/okgiybdbnlcpu23suvqq6lufze), but it needs a lot of VRAM, or you can run it through replicate, or here's the repo (https://github.com/Fanghua-Yu/SUPIR)

VRT

1 1,244 0.0 Python

VRT: A Video Restoration Transformer (official repository)
maxim

1 940 0.0 Python

[CVPR 2022 Oral] Official repository for "MAXIM: Multi-Axis MLP for Image Processing". SOTA for denoising, deblurring, deraining, dehazing, and enhancement.
maxim-pytorch

1 159 1.6 Python

[CVPR 2022 Oral] PyTorch re-implementation for "MAXIM: Multi-Axis MLP for Image Processing", with *training code*. Official Jax repo: https://github.com/google-research/maxim

Project mention: Need help to reduce vision transformer patch artifacts | /r/computervision | 2023-04-28

Hi there, i'm trying to do image enhancement task. Basically you give it some image and it makes it more colorful, vibrant, changes up some exposure and other things. I'm implementing it with Google Maxim architecture in pytorch. Unfortunately they dont include training code (original one is in JAX) and pyotorch only has the network, not training code.

InfluxDB

www.influxdata.com sponsored

Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).