├── .gitignore ├── README.md ├── checkpoints └── unet_adv │ ├── fE_86.pth │ ├── fI_86.pth │ └── fN_86.pth ├── dataset └── dataset.py ├── dataset_synthesis └── nyu_uw_syn_script.py ├── environment.yml ├── models ├── networks.py └── unet_parts.py ├── test.py ├── train.py └── visualize.py /.gitignore: -------------------------------------------------------------------------------- 1 | results/ -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/All-In-One-Underwater-Image-Enhancement-using-Domain-Adversarial-Learning/HEAD/README.md -------------------------------------------------------------------------------- /checkpoints/unet_adv/fE_86.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/All-In-One-Underwater-Image-Enhancement-using-Domain-Adversarial-Learning/HEAD/checkpoints/unet_adv/fE_86.pth -------------------------------------------------------------------------------- /checkpoints/unet_adv/fI_86.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/All-In-One-Underwater-Image-Enhancement-using-Domain-Adversarial-Learning/HEAD/checkpoints/unet_adv/fI_86.pth -------------------------------------------------------------------------------- /checkpoints/unet_adv/fN_86.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/All-In-One-Underwater-Image-Enhancement-using-Domain-Adversarial-Learning/HEAD/checkpoints/unet_adv/fN_86.pth -------------------------------------------------------------------------------- /dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/All-In-One-Underwater-Image-Enhancement-using-Domain-Adversarial-Learning/HEAD/dataset/dataset.py -------------------------------------------------------------------------------- /dataset_synthesis/nyu_uw_syn_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/All-In-One-Underwater-Image-Enhancement-using-Domain-Adversarial-Learning/HEAD/dataset_synthesis/nyu_uw_syn_script.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/All-In-One-Underwater-Image-Enhancement-using-Domain-Adversarial-Learning/HEAD/environment.yml -------------------------------------------------------------------------------- /models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/All-In-One-Underwater-Image-Enhancement-using-Domain-Adversarial-Learning/HEAD/models/networks.py -------------------------------------------------------------------------------- /models/unet_parts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/All-In-One-Underwater-Image-Enhancement-using-Domain-Adversarial-Learning/HEAD/models/unet_parts.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/All-In-One-Underwater-Image-Enhancement-using-Domain-Adversarial-Learning/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/All-In-One-Underwater-Image-Enhancement-using-Domain-Adversarial-Learning/HEAD/train.py -------------------------------------------------------------------------------- /visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/All-In-One-Underwater-Image-Enhancement-using-Domain-Adversarial-Learning/HEAD/visualize.py --------------------------------------------------------------------------------