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].