├── .gitignore ├── README.md ├── dataloaders ├── __init__.py ├── base_loader.py ├── paired_loader.py ├── single_loader.py └── unpaired_loader.py ├── datasets ├── __init__.py ├── base_dataset.py ├── single_dataset.py ├── superres_dataset.py └── unpaired_dataset.py ├── download_cyclegan_dataset.sh ├── models ├── CycleGAN_model.py ├── DeshadowNet_model.py ├── __init__.py ├── base_model.py ├── discriminators │ ├── __init__.py │ ├── cyclegan_discriminators.py │ ├── enlightengan_discriminators.py │ ├── maskshadowgan_discriminators.py │ └── srgan_discriminator.py ├── enlightengan_model.py ├── generators │ ├── __init__.py │ ├── cyclegan_generators.py │ ├── enlightengan_generators.py │ ├── maskshadowgan_generators.py │ └── srgan_generator.py ├── maskshadowgan_model.py └── srgan_model.py ├── options ├── __init__.py ├── base_options.py ├── cyclegan_options.py ├── deshadownet_options.py ├── enlightengan_options.py ├── maskshadowgan_options.py └── srgan_options.py ├── test ├── __init__.py ├── base_test.py ├── cyclegan_test.py ├── deshadownet_test.py ├── enlightengan_test.py ├── maskshadowgan_test.py └── srgan_test.py ├── train ├── __init__.py ├── base_train.py ├── cyclegan_train.py ├── deshadownet_train.py ├── enlightengan_train.py ├── maskshadowgan_train.py └── srgan_train.py └── utils ├── __init__.py ├── file_utils.py ├── im_utils.py ├── image_pool.py ├── mask_queue.py └── ops.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/README.md -------------------------------------------------------------------------------- /dataloaders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataloaders/base_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/dataloaders/base_loader.py -------------------------------------------------------------------------------- /dataloaders/paired_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/dataloaders/paired_loader.py -------------------------------------------------------------------------------- /dataloaders/single_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/dataloaders/single_loader.py -------------------------------------------------------------------------------- /dataloaders/unpaired_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/dataloaders/unpaired_loader.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/datasets/base_dataset.py -------------------------------------------------------------------------------- /datasets/single_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/datasets/single_dataset.py -------------------------------------------------------------------------------- /datasets/superres_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/datasets/superres_dataset.py -------------------------------------------------------------------------------- /datasets/unpaired_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/datasets/unpaired_dataset.py -------------------------------------------------------------------------------- /download_cyclegan_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/download_cyclegan_dataset.sh -------------------------------------------------------------------------------- /models/CycleGAN_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/CycleGAN_model.py -------------------------------------------------------------------------------- /models/DeshadowNet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/DeshadowNet_model.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/base_model.py -------------------------------------------------------------------------------- /models/discriminators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/discriminators/cyclegan_discriminators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/discriminators/cyclegan_discriminators.py -------------------------------------------------------------------------------- /models/discriminators/enlightengan_discriminators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/discriminators/enlightengan_discriminators.py -------------------------------------------------------------------------------- /models/discriminators/maskshadowgan_discriminators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/discriminators/maskshadowgan_discriminators.py -------------------------------------------------------------------------------- /models/discriminators/srgan_discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/discriminators/srgan_discriminator.py -------------------------------------------------------------------------------- /models/enlightengan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/enlightengan_model.py -------------------------------------------------------------------------------- /models/generators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/generators/cyclegan_generators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/generators/cyclegan_generators.py -------------------------------------------------------------------------------- /models/generators/enlightengan_generators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/generators/enlightengan_generators.py -------------------------------------------------------------------------------- /models/generators/maskshadowgan_generators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/generators/maskshadowgan_generators.py -------------------------------------------------------------------------------- /models/generators/srgan_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/generators/srgan_generator.py -------------------------------------------------------------------------------- /models/maskshadowgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/maskshadowgan_model.py -------------------------------------------------------------------------------- /models/srgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/models/srgan_model.py -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /options/base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/options/base_options.py -------------------------------------------------------------------------------- /options/cyclegan_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/options/cyclegan_options.py -------------------------------------------------------------------------------- /options/deshadownet_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/options/deshadownet_options.py -------------------------------------------------------------------------------- /options/enlightengan_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/options/enlightengan_options.py -------------------------------------------------------------------------------- /options/maskshadowgan_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/options/maskshadowgan_options.py -------------------------------------------------------------------------------- /options/srgan_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/options/srgan_options.py -------------------------------------------------------------------------------- /test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/base_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/test/base_test.py -------------------------------------------------------------------------------- /test/cyclegan_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/test/cyclegan_test.py -------------------------------------------------------------------------------- /test/deshadownet_test.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/enlightengan_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/test/enlightengan_test.py -------------------------------------------------------------------------------- /test/maskshadowgan_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/test/maskshadowgan_test.py -------------------------------------------------------------------------------- /test/srgan_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/test/srgan_test.py -------------------------------------------------------------------------------- /train/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train/base_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/train/base_train.py -------------------------------------------------------------------------------- /train/cyclegan_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/train/cyclegan_train.py -------------------------------------------------------------------------------- /train/deshadownet_train.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train/enlightengan_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/train/enlightengan_train.py -------------------------------------------------------------------------------- /train/maskshadowgan_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/train/maskshadowgan_train.py -------------------------------------------------------------------------------- /train/srgan_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/train/srgan_train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/utils/file_utils.py -------------------------------------------------------------------------------- /utils/im_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/utils/im_utils.py -------------------------------------------------------------------------------- /utils/image_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/utils/image_pool.py -------------------------------------------------------------------------------- /utils/mask_queue.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/utils/mask_queue.py -------------------------------------------------------------------------------- /utils/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wkhademi/ImageEnhancement/HEAD/utils/ops.py --------------------------------------------------------------------------------