├── LICENSE ├── README.md ├── diffiqa(r) ├── .gitignore ├── README.md ├── configs │ ├── inference_config.yaml │ ├── model_config.yaml │ └── train_config.yaml ├── dataset.py ├── inference.py ├── model_weights │ ├── .gitignore │ └── iresnet.py ├── train.py └── utils.py ├── diffiqa ├── .gitignore ├── README.md ├── bsrgan │ ├── degrade.py │ └── utils_image.py ├── configs │ ├── fr_config.yaml │ ├── inference_config.yaml │ └── train_config.yaml ├── dataset.py ├── diffusion_base │ ├── __init__.py │ ├── diff_dataset.py │ ├── diff_gauss.py │ ├── diff_main.py │ ├── diff_unet.py │ └── diff_util.py ├── inference.py ├── model_weights │ ├── .gitignore │ └── iresnet.py ├── train.py └── utils.py └── figures ├── diffiqa_edc.png ├── diffiqa_overview.png ├── diffiqa_pauc02.png ├── diffiqa_pauc03.png ├── diffiqa_rankings.png └── diffiqa_teaser.png /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/README.md -------------------------------------------------------------------------------- /diffiqa(r)/.gitignore: -------------------------------------------------------------------------------- 1 | 2 | *wandb 3 | *__pycache__ 4 | *nohup -------------------------------------------------------------------------------- /diffiqa(r)/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa(r)/README.md -------------------------------------------------------------------------------- /diffiqa(r)/configs/inference_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa(r)/configs/inference_config.yaml -------------------------------------------------------------------------------- /diffiqa(r)/configs/model_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa(r)/configs/model_config.yaml -------------------------------------------------------------------------------- /diffiqa(r)/configs/train_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa(r)/configs/train_config.yaml -------------------------------------------------------------------------------- /diffiqa(r)/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa(r)/dataset.py -------------------------------------------------------------------------------- /diffiqa(r)/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa(r)/inference.py -------------------------------------------------------------------------------- /diffiqa(r)/model_weights/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa(r)/model_weights/.gitignore -------------------------------------------------------------------------------- /diffiqa(r)/model_weights/iresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa(r)/model_weights/iresnet.py -------------------------------------------------------------------------------- /diffiqa(r)/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa(r)/train.py -------------------------------------------------------------------------------- /diffiqa(r)/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa(r)/utils.py -------------------------------------------------------------------------------- /diffiqa/.gitignore: -------------------------------------------------------------------------------- 1 | 2 | *wandb 3 | *__pycache__ 4 | *nohup -------------------------------------------------------------------------------- /diffiqa/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/README.md -------------------------------------------------------------------------------- /diffiqa/bsrgan/degrade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/bsrgan/degrade.py -------------------------------------------------------------------------------- /diffiqa/bsrgan/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/bsrgan/utils_image.py -------------------------------------------------------------------------------- /diffiqa/configs/fr_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/configs/fr_config.yaml -------------------------------------------------------------------------------- /diffiqa/configs/inference_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/configs/inference_config.yaml -------------------------------------------------------------------------------- /diffiqa/configs/train_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/configs/train_config.yaml -------------------------------------------------------------------------------- /diffiqa/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/dataset.py -------------------------------------------------------------------------------- /diffiqa/diffusion_base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/diffusion_base/__init__.py -------------------------------------------------------------------------------- /diffiqa/diffusion_base/diff_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/diffusion_base/diff_dataset.py -------------------------------------------------------------------------------- /diffiqa/diffusion_base/diff_gauss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/diffusion_base/diff_gauss.py -------------------------------------------------------------------------------- /diffiqa/diffusion_base/diff_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/diffusion_base/diff_main.py -------------------------------------------------------------------------------- /diffiqa/diffusion_base/diff_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/diffusion_base/diff_unet.py -------------------------------------------------------------------------------- /diffiqa/diffusion_base/diff_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/diffusion_base/diff_util.py -------------------------------------------------------------------------------- /diffiqa/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/inference.py -------------------------------------------------------------------------------- /diffiqa/model_weights/.gitignore: -------------------------------------------------------------------------------- 1 | 2 | diff_pretrain.pt 3 | weights.pth 4 | results -------------------------------------------------------------------------------- /diffiqa/model_weights/iresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/model_weights/iresnet.py -------------------------------------------------------------------------------- /diffiqa/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/train.py -------------------------------------------------------------------------------- /diffiqa/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/diffiqa/utils.py -------------------------------------------------------------------------------- /figures/diffiqa_edc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/figures/diffiqa_edc.png -------------------------------------------------------------------------------- /figures/diffiqa_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/figures/diffiqa_overview.png -------------------------------------------------------------------------------- /figures/diffiqa_pauc02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/figures/diffiqa_pauc02.png -------------------------------------------------------------------------------- /figures/diffiqa_pauc03.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/figures/diffiqa_pauc03.png -------------------------------------------------------------------------------- /figures/diffiqa_rankings.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/figures/diffiqa_rankings.png -------------------------------------------------------------------------------- /figures/diffiqa_teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSIbabnikz/DifFIQA/HEAD/figures/diffiqa_teaser.png --------------------------------------------------------------------------------