The polarization imaging technique leverages the disparity between target and background polarization information to mitigate the impact of backward scattered light, thereby enhancing image quality. However, the imaging model of this method exhibits limitations in extracting inter-image features, resulting in less-than-optimal outcomes in turbid underwater environments. In recent years, machine learning methodologies, particularly neural networks, have gained traction. These networks, renowned for their superior fitting capabilities, can effectively extract information from multiple images. The incorporation of an attention mechanism significantly augments the capacity of neural networks to extract inter-image correlation attributes, thereby mitigating the constraints of polarization imaging methods to a certain degree. To enhance the efficacy of polarization imaging in complex underwater environments, this paper introduces a super-resolution network with an integrated attention mechanism, termed as SRGAN-DP. This network is a fusion of an enhanced SRGAN network and the high-performance deep pyramidal split attention (DPSA) module, also proposed in this paper. SRGAN-DP is employed to perform high-resolution reconstruction of the underwater polarimetric image dataset, constructed specifically for this study. A comparative analysis with existing algorithms demonstrates that our proposed algorithm not only produces superior images but also exhibits robust performance in real-world environments.
Read full abstract