Dataset Preparation
Download UIEB Dataset and EUVP dataset
The uw_data directory consists of two main subdirectories: train, which contains subdirectories a (containing training input images) and b (containing corresponding ground truth images for training), and test, which contains subdirectories a (with testing input images) and b (containing corresponding ground truth images for testing).
To train the model, execute the following command in your terminal:
python train.py
If you found this work helpful, please cite it using the following reference:
@inproceedings{khan2023underwater, title={Underwater Image Enhancement with Phase Transfer and Attention}, author={Khan, MD Raqib and Kulkarni, Ashutosh and Phutke, Shruti S and Murala, Subrahmanyam}, booktitle={2023 International Joint Conference on Neural Networks (IJCNN)}, pages={1--8}, year={2023}, organization={IEEE}})
For any inquiries or further information, please contact me at srk689800@gmail.com.