├── .gitignore ├── LICENSE ├── README.md ├── configs ├── Real_ESRGAN │ ├── realesrgan_x2.yaml │ ├── realesrgan_x3.yaml │ ├── realesrgan_x4-finetune.yaml │ ├── realesrgan_x4.yaml │ ├── realesrnet_x2.yaml │ ├── realesrnet_x3.yaml │ ├── realesrnet_x4-finetune.yaml │ └── realesrnet_x4.yaml ├── degradation_model.yaml └── inference │ ├── images.yaml │ ├── large_images.yaml │ └── video.yaml ├── datasets └── README.md ├── demo ├── data │ ├── 00003.jpg │ ├── 0014.jpg │ └── 0030.jpg ├── inference_images.py ├── inference_large_images.py └── inference_video.py ├── docs └── training.md ├── figure ├── 00003_bicubic_x4.jpg ├── 00003_real_esrgan_x4.jpg ├── 0014_bicubic_x4.jpg ├── 0014_real_esrgan_x4.jpg ├── 0030_bicubic_x4.jpg └── 0030_real_esrgan_x4.jpg ├── pyproject.toml ├── real_esrgan ├── __init__.py ├── apis │ ├── __init__.py │ ├── large_image_super_resolution.py │ ├── super_resolution.py │ └── video_super_resolution.py ├── data │ ├── __init__.py │ ├── bicubic_image_dataset.py │ ├── degenerated_image_dataset.py │ ├── degradations.py │ ├── paired_image_dataset.py │ └── transforms.py ├── engine │ ├── __init__.py │ ├── backend.py │ ├── evaler.py │ └── trainer.py ├── evaluation │ ├── __init__.py │ └── metrics │ │ ├── __init__.py │ │ ├── mse.py │ │ ├── niqe.py │ │ ├── psnr.py │ │ └── ssim.py ├── layers │ ├── __init__.py │ ├── blocks.py │ └── ema.py ├── models │ ├── __init__.py │ ├── discriminator_for_unet.py │ ├── losses │ │ ├── __init__.py │ │ └── feature_loss.py │ ├── rrdbnet.py │ └── vgg_feature_extractor.py └── utils │ ├── __init__.py │ ├── checkpoint.py │ ├── color.py │ ├── diffjepg.py │ ├── envs.py │ ├── events.py │ ├── imgproc.py │ ├── matlab_functions.py │ ├── ops.py │ └── torch_utils.py ├── requirements.txt ├── results ├── .gitkeep └── pretrained_models │ ├── niqe_model.mat │ └── niqe_model.npz ├── scripts ├── get_df2k_datasets.sh └── get_test_datasets.sh ├── tests └── test_inference_time.py └── tools ├── eval.py ├── misc ├── generate_multi_scale_dataset.py └── slice_sub_image.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/README.md -------------------------------------------------------------------------------- /configs/Real_ESRGAN/realesrgan_x2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/configs/Real_ESRGAN/realesrgan_x2.yaml -------------------------------------------------------------------------------- /configs/Real_ESRGAN/realesrgan_x3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/configs/Real_ESRGAN/realesrgan_x3.yaml -------------------------------------------------------------------------------- /configs/Real_ESRGAN/realesrgan_x4-finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/configs/Real_ESRGAN/realesrgan_x4-finetune.yaml -------------------------------------------------------------------------------- /configs/Real_ESRGAN/realesrgan_x4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/configs/Real_ESRGAN/realesrgan_x4.yaml -------------------------------------------------------------------------------- /configs/Real_ESRGAN/realesrnet_x2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/configs/Real_ESRGAN/realesrnet_x2.yaml -------------------------------------------------------------------------------- /configs/Real_ESRGAN/realesrnet_x3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/configs/Real_ESRGAN/realesrnet_x3.yaml -------------------------------------------------------------------------------- /configs/Real_ESRGAN/realesrnet_x4-finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/configs/Real_ESRGAN/realesrnet_x4-finetune.yaml -------------------------------------------------------------------------------- /configs/Real_ESRGAN/realesrnet_x4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/configs/Real_ESRGAN/realesrnet_x4.yaml -------------------------------------------------------------------------------- /configs/degradation_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/configs/degradation_model.yaml -------------------------------------------------------------------------------- /configs/inference/images.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/configs/inference/images.yaml -------------------------------------------------------------------------------- /configs/inference/large_images.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/configs/inference/large_images.yaml -------------------------------------------------------------------------------- /configs/inference/video.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/configs/inference/video.yaml -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/datasets/README.md -------------------------------------------------------------------------------- /demo/data/00003.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/demo/data/00003.jpg -------------------------------------------------------------------------------- /demo/data/0014.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/demo/data/0014.jpg -------------------------------------------------------------------------------- /demo/data/0030.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/demo/data/0030.jpg -------------------------------------------------------------------------------- /demo/inference_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/demo/inference_images.py -------------------------------------------------------------------------------- /demo/inference_large_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/demo/inference_large_images.py -------------------------------------------------------------------------------- /demo/inference_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/demo/inference_video.py -------------------------------------------------------------------------------- /docs/training.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/docs/training.md -------------------------------------------------------------------------------- /figure/00003_bicubic_x4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/figure/00003_bicubic_x4.jpg -------------------------------------------------------------------------------- /figure/00003_real_esrgan_x4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/figure/00003_real_esrgan_x4.jpg -------------------------------------------------------------------------------- /figure/0014_bicubic_x4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/figure/0014_bicubic_x4.jpg -------------------------------------------------------------------------------- /figure/0014_real_esrgan_x4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/figure/0014_real_esrgan_x4.jpg -------------------------------------------------------------------------------- /figure/0030_bicubic_x4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/figure/0030_bicubic_x4.jpg -------------------------------------------------------------------------------- /figure/0030_real_esrgan_x4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/figure/0030_real_esrgan_x4.jpg -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/pyproject.toml -------------------------------------------------------------------------------- /real_esrgan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/__init__.py -------------------------------------------------------------------------------- /real_esrgan/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/apis/__init__.py -------------------------------------------------------------------------------- /real_esrgan/apis/large_image_super_resolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/apis/large_image_super_resolution.py -------------------------------------------------------------------------------- /real_esrgan/apis/super_resolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/apis/super_resolution.py -------------------------------------------------------------------------------- /real_esrgan/apis/video_super_resolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/apis/video_super_resolution.py -------------------------------------------------------------------------------- /real_esrgan/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/data/__init__.py -------------------------------------------------------------------------------- /real_esrgan/data/bicubic_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/data/bicubic_image_dataset.py -------------------------------------------------------------------------------- /real_esrgan/data/degenerated_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/data/degenerated_image_dataset.py -------------------------------------------------------------------------------- /real_esrgan/data/degradations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/data/degradations.py -------------------------------------------------------------------------------- /real_esrgan/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/data/paired_image_dataset.py -------------------------------------------------------------------------------- /real_esrgan/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/data/transforms.py -------------------------------------------------------------------------------- /real_esrgan/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/engine/__init__.py -------------------------------------------------------------------------------- /real_esrgan/engine/backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/engine/backend.py -------------------------------------------------------------------------------- /real_esrgan/engine/evaler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/engine/evaler.py -------------------------------------------------------------------------------- /real_esrgan/engine/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/engine/trainer.py -------------------------------------------------------------------------------- /real_esrgan/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/evaluation/__init__.py -------------------------------------------------------------------------------- /real_esrgan/evaluation/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/evaluation/metrics/__init__.py -------------------------------------------------------------------------------- /real_esrgan/evaluation/metrics/mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/evaluation/metrics/mse.py -------------------------------------------------------------------------------- /real_esrgan/evaluation/metrics/niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/evaluation/metrics/niqe.py -------------------------------------------------------------------------------- /real_esrgan/evaluation/metrics/psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/evaluation/metrics/psnr.py -------------------------------------------------------------------------------- /real_esrgan/evaluation/metrics/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/evaluation/metrics/ssim.py -------------------------------------------------------------------------------- /real_esrgan/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/layers/__init__.py -------------------------------------------------------------------------------- /real_esrgan/layers/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/layers/blocks.py -------------------------------------------------------------------------------- /real_esrgan/layers/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/layers/ema.py -------------------------------------------------------------------------------- /real_esrgan/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/models/__init__.py -------------------------------------------------------------------------------- /real_esrgan/models/discriminator_for_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/models/discriminator_for_unet.py -------------------------------------------------------------------------------- /real_esrgan/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/models/losses/__init__.py -------------------------------------------------------------------------------- /real_esrgan/models/losses/feature_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/models/losses/feature_loss.py -------------------------------------------------------------------------------- /real_esrgan/models/rrdbnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/models/rrdbnet.py -------------------------------------------------------------------------------- /real_esrgan/models/vgg_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/models/vgg_feature_extractor.py -------------------------------------------------------------------------------- /real_esrgan/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/utils/__init__.py -------------------------------------------------------------------------------- /real_esrgan/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/utils/checkpoint.py -------------------------------------------------------------------------------- /real_esrgan/utils/color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/utils/color.py -------------------------------------------------------------------------------- /real_esrgan/utils/diffjepg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/utils/diffjepg.py -------------------------------------------------------------------------------- /real_esrgan/utils/envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/utils/envs.py -------------------------------------------------------------------------------- /real_esrgan/utils/events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/utils/events.py -------------------------------------------------------------------------------- /real_esrgan/utils/imgproc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/utils/imgproc.py -------------------------------------------------------------------------------- /real_esrgan/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/utils/matlab_functions.py -------------------------------------------------------------------------------- /real_esrgan/utils/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/utils/ops.py -------------------------------------------------------------------------------- /real_esrgan/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/real_esrgan/utils/torch_utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | omegaconf 2 | opencv-python 3 | numpy 4 | typing 5 | tqdm 6 | scipy 7 | fvcore -------------------------------------------------------------------------------- /results/.gitkeep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /results/pretrained_models/niqe_model.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/results/pretrained_models/niqe_model.mat -------------------------------------------------------------------------------- /results/pretrained_models/niqe_model.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/results/pretrained_models/niqe_model.npz -------------------------------------------------------------------------------- /scripts/get_df2k_datasets.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/scripts/get_df2k_datasets.sh -------------------------------------------------------------------------------- /scripts/get_test_datasets.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/scripts/get_test_datasets.sh -------------------------------------------------------------------------------- /tests/test_inference_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/tests/test_inference_time.py -------------------------------------------------------------------------------- /tools/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/tools/eval.py -------------------------------------------------------------------------------- /tools/misc/generate_multi_scale_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/tools/misc/generate_multi_scale_dataset.py -------------------------------------------------------------------------------- /tools/misc/slice_sub_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/tools/misc/slice_sub_image.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lornatang/Real_ESRGAN-PyTorch/HEAD/tools/train.py --------------------------------------------------------------------------------