/traiNNer-redux-fork

(Fork of BasicSR) 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.

Primary LanguagePythonApache License 2.0Apache-2.0

A community fork of BasicSR and traiNNer-redux that adds experimental features.

  • Adds OmniSR network support.
  • Adds Contextual Loss, Color Loss, Averaging Downscale Loss
  • Adds Mix of Augmentations. Currently supports: CutBlur, CutMixUp, CutOut, CutMix, MixUp, RGB and blend.
  • Adds more optimizers. Currently supports: Lion, Tiger, Lamb, AdamP, Prodigy, DAdaptAdam and DAdaptAdan.

LICENSE

⚡HowTo | 🔧Installation | 💻Training Commands | 🐢DatasetPrepare | 🏰Model Zoo

📜 License and Acknowledgement

This project is released under the Apache 2.0 license.
More details about license and acknowledgement are in LICENSE.

This repository is a fork of BasicSR:

Xintao Wang, Liangbin Xie, Ke Yu, Kelvin C.K. Chan, Chen Change Loy and Chao Dong. BasicSR: Open Source Image and Video Restoration Toolbox. https://github.com/xinntao/BasicSR, 2022.