All Projects → WenlongZhang0517 → Ranksrgan

WenlongZhang0517 / Ranksrgan

ICCV 2019 (oral) RankSRGAN: Generative Adversarial Networks with Ranker for Image Super-Resolution. PyTorch implementation

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Ranksrgan

srgan
Pytorch implementation of "Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network"
Stars: ✭ 39 (-81.69%)
Mutual labels:  generative-adversarial-network, gan, super-resolution
Iseebetter
iSeeBetter: Spatio-Temporal Video Super Resolution using Recurrent-Generative Back-Projection Networks | Python3 | PyTorch | GANs | CNNs | ResNets | RNNs | Published in Springer Journal of Computational Visual Media, September 2020, Tsinghua University Press
Stars: ✭ 202 (-5.16%)
Mutual labels:  gan, generative-adversarial-network, super-resolution
Tensorflow Srgan
Tensorflow implementation of "Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network" (Ledig et al. 2017)
Stars: ✭ 33 (-84.51%)
Mutual labels:  gan, generative-adversarial-network, super-resolution
A Pytorch Tutorial To Super Resolution
Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network | a PyTorch Tutorial to Super-Resolution
Stars: ✭ 157 (-26.29%)
Mutual labels:  gan, generative-adversarial-network, super-resolution
DLSS
Deep Learning Super Sampling with Deep Convolutional Generative Adversarial Networks.
Stars: ✭ 88 (-58.69%)
Mutual labels:  generative-adversarial-network, gan, super-resolution
Pytorch Srgan
A modern PyTorch implementation of SRGAN
Stars: ✭ 289 (+35.68%)
Mutual labels:  gan, generative-adversarial-network, super-resolution
Awesome Gan For Medical Imaging
Awesome GAN for Medical Imaging
Stars: ✭ 1,814 (+751.64%)
Mutual labels:  gan, generative-adversarial-network, super-resolution
Anime Face Gan Keras
A DCGAN to generate anime faces using custom mined dataset
Stars: ✭ 161 (-24.41%)
Mutual labels:  gan, generative-adversarial-network
Tensorflow Mnist Gan Dcgan
Tensorflow implementation of Generative Adversarial Networks (GAN) and Deep Convolutional Generative Adversarial Netwokrs for MNIST dataset.
Stars: ✭ 163 (-23.47%)
Mutual labels:  gan, generative-adversarial-network
Gan Sandbox
Vanilla GAN implemented on top of keras/tensorflow enabling rapid experimentation & research. Branches correspond to implementations of stable GAN variations (i.e. ACGan, InfoGAN) and other promising variations of GANs like conditional and Wasserstein.
Stars: ✭ 210 (-1.41%)
Mutual labels:  gan, generative-adversarial-network
Gan2shape
Code for GAN2Shape (ICLR2021 oral)
Stars: ✭ 183 (-14.08%)
Mutual labels:  gan, generative-adversarial-network
Mmediting
OpenMMLab Image and Video Editing Toolbox
Stars: ✭ 2,618 (+1129.11%)
Mutual labels:  generative-adversarial-network, super-resolution
Image generator
DCGAN image generator 🖼️.
Stars: ✭ 173 (-18.78%)
Mutual labels:  gan, generative-adversarial-network
Dragan
A stable algorithm for GAN training
Stars: ✭ 189 (-11.27%)
Mutual labels:  gan, generative-adversarial-network
Frontalization
Pytorch deep learning face frontalization model
Stars: ✭ 160 (-24.88%)
Mutual labels:  gan, generative-adversarial-network
Gannotation
GANnotation (PyTorch): Landmark-guided face to face synthesis using GANs (And a triple consistency loss!)
Stars: ✭ 167 (-21.6%)
Mutual labels:  gan, generative-adversarial-network
Facegan
TF implementation of our ECCV 2018 paper: Semi-supervised Adversarial Learning to Generate Photorealistic Face Images of New Identities from 3D Morphable Model
Stars: ✭ 176 (-17.37%)
Mutual labels:  gan, generative-adversarial-network
Freezed
Freeze the Discriminator: a Simple Baseline for Fine-Tuning GANs (CVPRW 2020)
Stars: ✭ 195 (-8.45%)
Mutual labels:  gan, generative-adversarial-network
Creative Adversarial Networks
(WIP) Implementation of Creative Adversarial Networks https://arxiv.org/pdf/1706.07068.pdf
Stars: ✭ 193 (-9.39%)
Mutual labels:  gan, generative-adversarial-network
Paddlegan
PaddlePaddle GAN library, including lots of interesting applications like First-Order motion transfer, wav2lip, picture repair, image editing, photo2cartoon, image style transfer, and so on.
Stars: ✭ 4,987 (+2241.31%)
Mutual labels:  gan, super-resolution

RankSRGAN

Paper | Supplementary file | Project Page

RankSRGAN: Generative Adversarial Networks with Ranker for Image Super-Resolution

By Wenlong Zhang, Yihao Liu, Chao Dong, Yu Qiao


Dependencies

Codes

  • We update the codes version based on mmsr. The old version can be downloaded from Google Drive
  • This version is under testing. We will provide more details of RankSRGAN in near future.

How to Test

  1. Clone this github repo.
git clone https://github.com/WenlongZhang0724/RankSRGAN.git
cd RankSRGAN
  1. Place your own low-resolution images in ./LR folder.
  2. Download pretrained models from Google Drive. Place the models in ./experiments/pretrained_models/. We provide three Ranker models and three RankSRGAN models (see model list).
  3. Run test. We provide RankSRGAN (NIQE, Ma, PI) model and you can config in the test.py.
python test.py -opt options/test/test_RankSRGAN.yml
  1. The results are in ./results folder.

How to Train

Train Ranker

  1. Download DIV2K and Flickr2K from Google Drive or Baidu Drive
  2. Generate rank dataset ./datasets/generate_rankdataset/
  3. Run command:
python train_rank.py -opt options/train/train_Ranker.yml

Train RankSRGAN

We use a PSNR-oriented pretrained SR model to initialize the parameters for better quality.

  1. Prepare datasets, usually the DIV2K dataset.
  2. Prerapre the PSNR-oriented pretrained model. You can use the mmsr_SRResNet_pretrain.pth as the pretrained model that can be downloaded from Google Drive.
  3. Modify the configuration file options/train/train_RankSRGAN.json
  4. Run command:
python train.py -opt options/train/train_RankSRGAN.yml

or

python train_niqe.py -opt options/train/train_RankSRGAN.yml

Using the train.py can output the convergence curves with PSNR; Using the train_niqe.py can output the convergence curves with NIQE and PSNR.

Acknowledgement

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].