Improved LYT-Net Image Enhancement Model for Underwater Robotics Vision

Although underwater robots can replace humans to explore the ocean which is rich in resources but fraught with unknown risks, there are phenomena such as monotonous colors, complex backgrounds and uneven illumination in the underwater environment, which makes it difficult for underwater robots relying on the optical vision system to obtain clear images and is not conducive to subsequent recognition and grasping tasks. In this paper, relevant preparatory work was carried out for underwater image processing applicable to underwater robots, and the underwater imaging model as well as other underwater image processing models were analyzed. Subsequently, a method combining the underwater physical model and the dark channel image was proposed to improve LYT-Net, enabling it to be applicable to mobile embedded devices such as underwater robots. After training and testing on the real datasets LSUI and UIEB, our model achieved good results on important indicators such as PSNR and SSIM. Meanwhile, it met the theoretical requirements of mobile devices in terms of FLOPs, Params and single-frame processing time, which verified its robustness and real-time processing ability in various underwater environments.