This is an implement of the TaterFlow, “HUPE: Heuristic Underwater Perceptual Enhancement with Semantic Collaborative Learning", Zengxi Zhang, Zhiying Jiang, Long Ma, Jinyuan Liu, Xin Fan, Risheng Liu*, 2024, under review.

Download the pre-trained model and put it in experiments/

• Baidu Yun
  code: ​ 1ecu ​
• Google Drive

To test the pre-trained models for Underwater Enhancement on example images, run

cd code
python test.py

If you find our work useful in your research, please cite our conference version:

@inproceedings{zhang2023waterflow,
	title={WaterFlow: Heuristic Normalizing Flow for Underwater Image Enhancement and Beyond},
	author={Zhang, Zengxi and Jiang, Zhiying and Liu, Jinyuan and Fan, Xin and Liu, Risheng},
	booktitle={Proceedings of the 31st ACM International Conference on Multimedia},
	pages={7314--7323},
	year={2023}
}

Should you have any question, please contact [Zengxi Zhang].