├── ReadMe.md ├── archs ├── RRDBNet.py ├── blocks.py ├── hinet_arch.py ├── nafnet.py └── nafnet_util.py ├── data_augmentation.py ├── dataset.py ├── environment.yaml ├── image_tools.py ├── inference_UGPNet.py ├── loss ├── Lpips.py ├── PSNRloss.py ├── __init__.py ├── contextual.py ├── gan_loss.py └── vgg.py ├── models ├── UGPNet.py ├── fusmodule.py ├── resmodule.py ├── stylegan2.py └── synmodule.py ├── results ├── nafnet_deblur │ └── inference_UGPNet.py ├── nafnet_denoise │ └── inference_UGPNet.py └── rrdb_x8 │ └── inference_UGPNet.py ├── test_scripts ├── test_UGPNet_w_NAFNet_deblur.sh ├── test_UGPNet_w_NAFNet_denoise.sh └── test_UGPNet_w_RRDBNet_sr.sh ├── train_fusmodule.py ├── train_resmodule.py ├── train_synmodule.py └── utils.py /ReadMe.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/ReadMe.md -------------------------------------------------------------------------------- /archs/RRDBNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/archs/RRDBNet.py -------------------------------------------------------------------------------- /archs/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/archs/blocks.py -------------------------------------------------------------------------------- /archs/hinet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/archs/hinet_arch.py -------------------------------------------------------------------------------- /archs/nafnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/archs/nafnet.py -------------------------------------------------------------------------------- /archs/nafnet_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/archs/nafnet_util.py -------------------------------------------------------------------------------- /data_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/data_augmentation.py -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/dataset.py -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/environment.yaml -------------------------------------------------------------------------------- /image_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/image_tools.py -------------------------------------------------------------------------------- /inference_UGPNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/inference_UGPNet.py -------------------------------------------------------------------------------- /loss/Lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/loss/Lpips.py -------------------------------------------------------------------------------- /loss/PSNRloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/loss/PSNRloss.py -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /loss/contextual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/loss/contextual.py -------------------------------------------------------------------------------- /loss/gan_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/loss/gan_loss.py -------------------------------------------------------------------------------- /loss/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/loss/vgg.py -------------------------------------------------------------------------------- /models/UGPNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/models/UGPNet.py -------------------------------------------------------------------------------- /models/fusmodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/models/fusmodule.py -------------------------------------------------------------------------------- /models/resmodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/models/resmodule.py -------------------------------------------------------------------------------- /models/stylegan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/models/stylegan2.py -------------------------------------------------------------------------------- /models/synmodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/models/synmodule.py -------------------------------------------------------------------------------- /results/nafnet_deblur/inference_UGPNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/results/nafnet_deblur/inference_UGPNet.py -------------------------------------------------------------------------------- /results/nafnet_denoise/inference_UGPNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/results/nafnet_denoise/inference_UGPNet.py -------------------------------------------------------------------------------- /results/rrdb_x8/inference_UGPNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/results/rrdb_x8/inference_UGPNet.py -------------------------------------------------------------------------------- /test_scripts/test_UGPNet_w_NAFNet_deblur.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/test_scripts/test_UGPNet_w_NAFNet_deblur.sh -------------------------------------------------------------------------------- /test_scripts/test_UGPNet_w_NAFNet_denoise.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/test_scripts/test_UGPNet_w_NAFNet_denoise.sh -------------------------------------------------------------------------------- /test_scripts/test_UGPNet_w_RRDBNet_sr.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/test_scripts/test_UGPNet_w_RRDBNet_sr.sh -------------------------------------------------------------------------------- /train_fusmodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/train_fusmodule.py -------------------------------------------------------------------------------- /train_resmodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/train_resmodule.py -------------------------------------------------------------------------------- /train_synmodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/train_synmodule.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hwayoon2/UGPNet/HEAD/utils.py --------------------------------------------------------------------------------