https://github.com/patil-suraj/swinir-jax

JAX implementation of SwinIR model.
https://github.com/patil-suraj/swinir-jax

Last synced: 1 day ago
JSON representation

JAX implementation of SwinIR model.

• Host: GitHub
• URL: https://github.com/patil-suraj/swinir-jax
• Owner: patil-suraj
• License: mit
• Created: 2022-08-10T09:53:43.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2022-10-24T16:56:25.000Z (about 2 years ago)
• Last Synced: 2024-11-04T05:33:31.296Z (2 months ago)
• Language: Jupyter Notebook
• Size: 7.8 MB
• Stars: 4
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE.txt

Awesome Lists containing this project

README

        
# SwinIR-JAX

Unofficial JAX implementation of [SwinIR](https://github.com/JingyunLiang/SwinIR).

As of now, this repo only supports the Real SR task from SwinIR, pull requests are more than welcome for other tasks. 

## How to use



The checkpoints are available on the [huggingface hub](https://huggingface.co/models?other=swin-ir).

## Citation

    @article{liang2021swinir,

      title={SwinIR: Image Restoration Using Swin Transformer},

      author={Liang, Jingyun and Cao, Jiezhang and Sun, Guolei and Zhang, Kai and Van Gool, Luc and Timofte, Radu},

      journal={arXiv preprint arXiv:2108.10257},

      year={2021}

    }

## Acknowledgement

Some of the JAX code is adapted from the amazing JAX implementation of `SwinTransformer` from [jax-models](https://github.com/DarshanDeshpande/jax-models) repo by @DarshanDeshpande