├── .gitignore ├── LICENSE ├── NIQE_Script ├── matlab_utils.py ├── metric_util.py ├── niqe.py └── niqe_pris_params.npz ├── README.md ├── Training.md ├── Visualization_Script ├── Multi_discriminator_Output.py ├── combine.py ├── discriminator_attention_visual(Multi).py └── discriminator_attention_visual(Single).py ├── aesrgan ├── __init__.py ├── archs │ ├── __init__.py │ ├── discriminator_arch.py │ ├── generator_arch.py │ └── multiscale_arch.py ├── data │ ├── __init__.py │ ├── realesrgan_dataset.py │ └── realesrgan_paired_dataset.py ├── models │ ├── __init__.py │ ├── aesrgan_model.py │ └── realesrnet_model.py ├── train.py ├── utils.py └── weights │ └── README.md ├── docs └── ncnn_conversion.md ├── figures ├── AttentionBlock.png ├── AttentionUnet.png ├── AttentionVisual.png ├── Compare.png ├── Discard Distortion.png ├── Example Images.png ├── GAN.png ├── Layers.png ├── Metric.png ├── Sharper.png ├── TotalArch.png ├── UnetArch.png ├── multi1.png ├── multi2.png └── unet_output.png ├── inference_aesrgan.py ├── inputs ├── 00003.png ├── 0014.jpg ├── 1.png ├── ADE_val_00000114.jpg ├── OST_009.png ├── img_003_SRF_4_HR_out.png ├── img_015_SRF_4_HR.png ├── img_081_SRF_4_HR_out.png ├── tree_alpha_16bit.png └── wolf_gray.jpg ├── options ├── finetune_multiaesrgan_x4plus.yml ├── train_aesrgan_x4plus.yml ├── train_multiaesrgan_x4plus.yml └── train_realesrnet_x4plus.yml ├── requirements.txt ├── scripts ├── data_preprocess │ ├── extract_subimages.py │ ├── meta_info_generate.py │ └── multiscale_generate.py ├── extract_subimages.py ├── generate_meta_info.py ├── generate_meta_info_pairdata.py ├── generate_multiscale_DF2K.py └── pytorch2onnx.py ├── setup.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/LICENSE -------------------------------------------------------------------------------- /NIQE_Script/matlab_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/NIQE_Script/matlab_utils.py -------------------------------------------------------------------------------- /NIQE_Script/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/NIQE_Script/metric_util.py -------------------------------------------------------------------------------- /NIQE_Script/niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/NIQE_Script/niqe.py -------------------------------------------------------------------------------- /NIQE_Script/niqe_pris_params.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/NIQE_Script/niqe_pris_params.npz -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/README.md -------------------------------------------------------------------------------- /Training.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/Training.md -------------------------------------------------------------------------------- /Visualization_Script/Multi_discriminator_Output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/Visualization_Script/Multi_discriminator_Output.py -------------------------------------------------------------------------------- /Visualization_Script/combine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/Visualization_Script/combine.py -------------------------------------------------------------------------------- /Visualization_Script/discriminator_attention_visual(Multi).py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/Visualization_Script/discriminator_attention_visual(Multi).py -------------------------------------------------------------------------------- /Visualization_Script/discriminator_attention_visual(Single).py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/Visualization_Script/discriminator_attention_visual(Single).py -------------------------------------------------------------------------------- /aesrgan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/__init__.py -------------------------------------------------------------------------------- /aesrgan/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/archs/__init__.py -------------------------------------------------------------------------------- /aesrgan/archs/discriminator_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/archs/discriminator_arch.py -------------------------------------------------------------------------------- /aesrgan/archs/generator_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/archs/generator_arch.py -------------------------------------------------------------------------------- /aesrgan/archs/multiscale_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/archs/multiscale_arch.py -------------------------------------------------------------------------------- /aesrgan/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/data/__init__.py -------------------------------------------------------------------------------- /aesrgan/data/realesrgan_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/data/realesrgan_dataset.py -------------------------------------------------------------------------------- /aesrgan/data/realesrgan_paired_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/data/realesrgan_paired_dataset.py -------------------------------------------------------------------------------- /aesrgan/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/models/__init__.py -------------------------------------------------------------------------------- /aesrgan/models/aesrgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/models/aesrgan_model.py -------------------------------------------------------------------------------- /aesrgan/models/realesrnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/models/realesrnet_model.py -------------------------------------------------------------------------------- /aesrgan/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/train.py -------------------------------------------------------------------------------- /aesrgan/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/utils.py -------------------------------------------------------------------------------- /aesrgan/weights/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/aesrgan/weights/README.md -------------------------------------------------------------------------------- /docs/ncnn_conversion.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/docs/ncnn_conversion.md -------------------------------------------------------------------------------- /figures/AttentionBlock.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/AttentionBlock.png -------------------------------------------------------------------------------- /figures/AttentionUnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/AttentionUnet.png -------------------------------------------------------------------------------- /figures/AttentionVisual.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/AttentionVisual.png -------------------------------------------------------------------------------- /figures/Compare.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/Compare.png -------------------------------------------------------------------------------- /figures/Discard Distortion.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/Discard Distortion.png -------------------------------------------------------------------------------- /figures/Example Images.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/Example Images.png -------------------------------------------------------------------------------- /figures/GAN.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/GAN.png -------------------------------------------------------------------------------- /figures/Layers.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/Layers.png -------------------------------------------------------------------------------- /figures/Metric.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/Metric.png -------------------------------------------------------------------------------- /figures/Sharper.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/Sharper.png -------------------------------------------------------------------------------- /figures/TotalArch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/TotalArch.png -------------------------------------------------------------------------------- /figures/UnetArch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/UnetArch.png -------------------------------------------------------------------------------- /figures/multi1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/multi1.png -------------------------------------------------------------------------------- /figures/multi2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/multi2.png -------------------------------------------------------------------------------- /figures/unet_output.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/figures/unet_output.png -------------------------------------------------------------------------------- /inference_aesrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/inference_aesrgan.py -------------------------------------------------------------------------------- /inputs/00003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/inputs/00003.png -------------------------------------------------------------------------------- /inputs/0014.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/inputs/0014.jpg -------------------------------------------------------------------------------- /inputs/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/inputs/1.png -------------------------------------------------------------------------------- /inputs/ADE_val_00000114.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/inputs/ADE_val_00000114.jpg -------------------------------------------------------------------------------- /inputs/OST_009.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/inputs/OST_009.png -------------------------------------------------------------------------------- /inputs/img_003_SRF_4_HR_out.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/inputs/img_003_SRF_4_HR_out.png -------------------------------------------------------------------------------- /inputs/img_015_SRF_4_HR.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/inputs/img_015_SRF_4_HR.png -------------------------------------------------------------------------------- /inputs/img_081_SRF_4_HR_out.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/inputs/img_081_SRF_4_HR_out.png -------------------------------------------------------------------------------- /inputs/tree_alpha_16bit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/inputs/tree_alpha_16bit.png -------------------------------------------------------------------------------- /inputs/wolf_gray.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/inputs/wolf_gray.jpg -------------------------------------------------------------------------------- /options/finetune_multiaesrgan_x4plus.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/options/finetune_multiaesrgan_x4plus.yml -------------------------------------------------------------------------------- /options/train_aesrgan_x4plus.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/options/train_aesrgan_x4plus.yml -------------------------------------------------------------------------------- /options/train_multiaesrgan_x4plus.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/options/train_multiaesrgan_x4plus.yml -------------------------------------------------------------------------------- /options/train_realesrnet_x4plus.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/options/train_realesrnet_x4plus.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/data_preprocess/extract_subimages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/scripts/data_preprocess/extract_subimages.py -------------------------------------------------------------------------------- /scripts/data_preprocess/meta_info_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/scripts/data_preprocess/meta_info_generate.py -------------------------------------------------------------------------------- /scripts/data_preprocess/multiscale_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/scripts/data_preprocess/multiscale_generate.py -------------------------------------------------------------------------------- /scripts/extract_subimages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/scripts/extract_subimages.py -------------------------------------------------------------------------------- /scripts/generate_meta_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/scripts/generate_meta_info.py -------------------------------------------------------------------------------- /scripts/generate_meta_info_pairdata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/scripts/generate_meta_info_pairdata.py -------------------------------------------------------------------------------- /scripts/generate_multiscale_DF2K.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/scripts/generate_multiscale_DF2K.py -------------------------------------------------------------------------------- /scripts/pytorch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/scripts/pytorch2onnx.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/setup.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stroking-fishes-ml-corp/A-ESRGAN/HEAD/train.py --------------------------------------------------------------------------------