HIFI-Net: A Novel Network for Enhancement to Underwater Optical Images

A novel network for enhancement of underwater optical images is proposed in this paper. It contains a Reinforcement Fusion Module for Haar wavelet (RFM-Haar) images. Fusion is performed to enrich available underlying information for better enhancement. As this network transform Haar Images into Fusion Images, it is called HIFI-Net. The experimental results based on Mean Square Error (0.1952, 0.1512 and 0.4683), Peak Signal to Noise Ratio (25.2241, 26.3342 and 21.4253) and Structure Similarity Index Measure (0.8265, 0.8819 and 0.8012) on three public datasets (EUVP, UFO-110 and UIEB) demonstrate that the proposed HIFI-Net performs better than some existing state-of-the-art methods.