SwinIR-JAX
Unofficial JAX implementation of 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.
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 repo by @DarshanDeshpande