Optimization color correction and regression model-based underwater image restoration method

Abstract

The present invention discloses an optimization color correction and regression model-based underwater image restoration method. The method includes the following steps that: the color deviation of an underwater image is removed by adopting an optimization theory-based color correction algorithm; the global background light of the underwater image is estimated based on quadtree decomposition and optical properties; a training sample is synthesized, characteristics related to absorption functions are designed, and a regression model is trained; the trained regression model is used to estimate the absorption function of a red channel, a distance between a camera and a scene point is determined, and the absorption function of a blue channel can be estimated; and a clear underwater image is restored according to an underwater optical imaging model. Compared with the prior art, the optimization color correction and regression model-based underwater image restoration method of the present invention can better restore the true color of the underwater image; a global background light estimation method can effectively remove the influences of bright objects and suspended particles in the water, and therefore, the optimization color correction and regression model-based underwater image restoration method has higher accuracy and higher robustness; and a learning-based framework is adopted for the first time to estimate the absorption functions of the underwater imaging model, so that the method of the invention has better robustness.

Description

Technical field [0001] The invention relates to the technical field of image processing, and proposes an underwater image restoration method based on an optimized color correction and regression model. Background technique [0002] With the rapid population expansion, the depletion of land resources, and the deteriorating environment, the development of marine resources is a far-reaching strategic choice facing the 21st century. Therefore, the importance of theories and technologies such as the acquisition, transmission and processing of marine information has become more prominent. The visual quality of underwater images plays an extremely important role in marine engineering applications and scientific research, such as underwater ecological research, marine rescue, underwater oil pipeline leakage monitoring and other application scenarios. Due to the effects of special light absorption and scattering, images taken underwater often have shortcomings such as low contrast, limit...

AI Technical Summary

Problems solved by technology

[0004] The present invention proposes an underwater image restoration method based on the optimal color correction and regression model. By comprehensively utilizing the optimization theory and the regression model, the color deviation of the underwater image can be effectively corrected, and the problem of low contrast and definition and texture problems can be solved. Underwater image restoration method for problems such as loss of details and

Images