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 [email protected].