├── .gitignore ├── README.md ├── data ├── __init__.py ├── aligned_dataset.py ├── base_dataset.py ├── cataract_dataset.py ├── cataract_guide_padding_dataset.py ├── cataract_with_mask_dataset.py ├── get_low_quality │ ├── __init__.py │ ├── cataract_simulation.py │ ├── degrad_de.py │ ├── fundus_prep.py │ ├── main_degradation.py │ ├── run_pre_process.py │ └── utils_de.py └── image_folder.py ├── images ├── Output.png ├── arcnet.png ├── arcnet_overview.png ├── cataract_dataset │ ├── source │ │ └── 04_test0.jpg │ ├── source_mask │ │ └── 04_test0.png │ ├── target │ │ ├── 2A.jpg │ │ └── cataract_003.png │ └── target_mask │ │ ├── 2A.png │ │ └── cataract_003.png ├── gfenet_overview.png ├── original_image │ ├── 01_test.tif │ ├── 02_test.tif │ └── 03_test.tif └── scrnet_overview.png ├── models ├── __init__.py ├── arcnet_model.py ├── backbone │ ├── gfenet_backbone.py │ └── unet_combine_2layer.py ├── base_model.py ├── cycle_gan_model.py ├── gfenet_model.py ├── guided_filter_pytorch │ ├── HFC_filter.py │ ├── __init__.py │ ├── box_filter.py │ ├── guided_filter.py │ ├── guided_filter_attention.py │ └── sobel_filter.py ├── networks.py ├── pix2pix_model.py ├── pixDA_sobel_model.py └── scrnet_model.py ├── options ├── __init__.py ├── base_options.py ├── test_options.py └── train_options.py ├── requirements.txt ├── test.py ├── train.py └── util ├── __init__.py ├── cataract_simulation.py ├── get_data.py ├── get_mask.py ├── html.py ├── image_pool.py ├── util.py └── visualizer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/aligned_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/aligned_dataset.py -------------------------------------------------------------------------------- /data/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/base_dataset.py -------------------------------------------------------------------------------- /data/cataract_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/cataract_dataset.py -------------------------------------------------------------------------------- /data/cataract_guide_padding_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/cataract_guide_padding_dataset.py -------------------------------------------------------------------------------- /data/cataract_with_mask_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/cataract_with_mask_dataset.py -------------------------------------------------------------------------------- /data/get_low_quality/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/get_low_quality/__init__.py -------------------------------------------------------------------------------- /data/get_low_quality/cataract_simulation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/get_low_quality/cataract_simulation.py -------------------------------------------------------------------------------- /data/get_low_quality/degrad_de.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/get_low_quality/degrad_de.py -------------------------------------------------------------------------------- /data/get_low_quality/fundus_prep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/get_low_quality/fundus_prep.py -------------------------------------------------------------------------------- /data/get_low_quality/main_degradation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/get_low_quality/main_degradation.py -------------------------------------------------------------------------------- /data/get_low_quality/run_pre_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/get_low_quality/run_pre_process.py -------------------------------------------------------------------------------- /data/get_low_quality/utils_de.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/get_low_quality/utils_de.py -------------------------------------------------------------------------------- /data/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/data/image_folder.py -------------------------------------------------------------------------------- /images/Output.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/Output.png -------------------------------------------------------------------------------- /images/arcnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/arcnet.png -------------------------------------------------------------------------------- /images/arcnet_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/arcnet_overview.png -------------------------------------------------------------------------------- /images/cataract_dataset/source/04_test0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/cataract_dataset/source/04_test0.jpg -------------------------------------------------------------------------------- /images/cataract_dataset/source_mask/04_test0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/cataract_dataset/source_mask/04_test0.png -------------------------------------------------------------------------------- /images/cataract_dataset/target/2A.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/cataract_dataset/target/2A.jpg -------------------------------------------------------------------------------- /images/cataract_dataset/target/cataract_003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/cataract_dataset/target/cataract_003.png -------------------------------------------------------------------------------- /images/cataract_dataset/target_mask/2A.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/cataract_dataset/target_mask/2A.png -------------------------------------------------------------------------------- /images/cataract_dataset/target_mask/cataract_003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/cataract_dataset/target_mask/cataract_003.png -------------------------------------------------------------------------------- /images/gfenet_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/gfenet_overview.png -------------------------------------------------------------------------------- /images/original_image/01_test.tif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/original_image/01_test.tif -------------------------------------------------------------------------------- /images/original_image/02_test.tif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/original_image/02_test.tif -------------------------------------------------------------------------------- /images/original_image/03_test.tif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/original_image/03_test.tif -------------------------------------------------------------------------------- /images/scrnet_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/images/scrnet_overview.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/arcnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/arcnet_model.py -------------------------------------------------------------------------------- /models/backbone/gfenet_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/backbone/gfenet_backbone.py -------------------------------------------------------------------------------- /models/backbone/unet_combine_2layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/backbone/unet_combine_2layer.py -------------------------------------------------------------------------------- /models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/base_model.py -------------------------------------------------------------------------------- /models/cycle_gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/cycle_gan_model.py -------------------------------------------------------------------------------- /models/gfenet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/gfenet_model.py -------------------------------------------------------------------------------- /models/guided_filter_pytorch/HFC_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/guided_filter_pytorch/HFC_filter.py -------------------------------------------------------------------------------- /models/guided_filter_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/guided_filter_pytorch/box_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/guided_filter_pytorch/box_filter.py -------------------------------------------------------------------------------- /models/guided_filter_pytorch/guided_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/guided_filter_pytorch/guided_filter.py -------------------------------------------------------------------------------- /models/guided_filter_pytorch/guided_filter_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/guided_filter_pytorch/guided_filter_attention.py -------------------------------------------------------------------------------- /models/guided_filter_pytorch/sobel_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/guided_filter_pytorch/sobel_filter.py -------------------------------------------------------------------------------- /models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/networks.py -------------------------------------------------------------------------------- /models/pix2pix_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/pix2pix_model.py -------------------------------------------------------------------------------- /models/pixDA_sobel_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/pixDA_sobel_model.py -------------------------------------------------------------------------------- /models/scrnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/models/scrnet_model.py -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/options/__init__.py -------------------------------------------------------------------------------- /options/base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/options/base_options.py -------------------------------------------------------------------------------- /options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/options/test_options.py -------------------------------------------------------------------------------- /options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/options/train_options.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/train.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/util/__init__.py -------------------------------------------------------------------------------- /util/cataract_simulation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/util/cataract_simulation.py -------------------------------------------------------------------------------- /util/get_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/util/get_data.py -------------------------------------------------------------------------------- /util/get_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/util/get_mask.py -------------------------------------------------------------------------------- /util/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/util/html.py -------------------------------------------------------------------------------- /util/image_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/util/image_pool.py -------------------------------------------------------------------------------- /util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/util/util.py -------------------------------------------------------------------------------- /util/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liamheng/Annotation-free-Fundus-Image-Enhancement/HEAD/util/visualizer.py --------------------------------------------------------------------------------