A Two-Stage Underwater Enhancement Network Based on Structure Decomposition and Characteristics of Underwater Imaging

The trained model of the paper S. Wu, T. Luo, G. Jiang, M. Yu, H. Xu, Z. Zhu and Y. Song, "A Two-Stage Underwater Enhancement Network Based on Structure Decomposition and Characteristics of Underwater Imaging," IEEE Journal of Oceanic Engineering, doi: 10.1109/JOE.2021.3064093.

Usage

Requirements

Matlab

Python = 3.6

TensorFlow =1.9.0

Numpy = 1.14.3

OpenCV-Python = 3.4.1

Testing

1. Clone the repo
2. Put the test underwater images into the "Inputs" folder
3. Run main.m in the "DCT" folder to get HF and LF
4. Run test.py
5. Find enhanced underwater images in the "Outputs" folder

Contact Us

If you have any questions, please contact us (xuhaiyong@nbu.edu.cn).