Underwater image restoration via non-convex non-smooth variation and thermal exchange optimization

The quality of underwater images is an important problem for resource detection. However, the light scattering and plankton in water can impact the quality of underwater images. In this paper, a novel underwater image restoration based on non-convex, non-smooth variation and thermal exchange optimization is proposed. Firstly, the underwater dark channel prior is used to estimate the rough transmission map. Secondly, the rough transmission map is refined by the proposed adaptive non-convex non-smooth variation. Then, Thermal Exchange Optimization is applied to compensate for the red channel of underwater images. Finally, the restored image can be estimated via the image formation model. The results show that the proposed algorithm can output high-quality images, according to qualitative and quantitative analysis.