Affiliation:
1. College of Communication and Information Technology, Xi’an University of Science and Technology, Xi’an 710600, China
2. Xi’an Key Laboratory of Network Convergence Communication, Xi’an University of Science and Technology, Xi’an 710054, China
3. College of Computer Science and Technology, Xi’an University of Science and Technology, Xi’an 710600, China
Abstract
Aiming at the problems of incomplete dehazing, color distortion, and loss of detail and edge information encountered by existing algorithms when processing images of underground coal mines, an image dehazing algorithm for underground coal mines, named CAB CA DSConv Fusion gUNet (CCDF-gUNet), is proposed. First, Dynamic Snake Convolution (DSConv) is introduced to replace traditional convolutions, enhancing the feature extraction capability. Second, residual attention convolution blocks are constructed to simultaneously focus on both local and global information in images. Additionally, the Coordinate Attention (CA) module is utilized to learn the coordinate information of features so that the model can better capture the key information in images. Furthermore, to simultaneously focus on the detail and structural consistency of images, a fusion loss function is introduced. Finally, based on the test verification of the public dataset Haze-4K, the Peak Signal-to-Noise Ratio (PSNR), Structural Similarity (SSIM), and Mean Squared Error (MSE) are 30.72 dB, 0.976, and 55.04, respectively, and on a self-made underground coal mine dataset, they are 31.18 dB, 0.971, and 49.66, respectively. The experimental results show that the algorithm performs well in dehazing, effectively avoids color distortion, and retains image details and edge information, providing some theoretical references for image processing in coal mine surveillance videos.
Funder
the Shaanxi Provincial Science and Technology Plan Project
Reference48 articles.
1. McCartney, E.J. (1976). Optics of the Atmosphere: Scattering by Molecules and Particles, Wiley.
2. Nayar, S.K., and Narasimhan, S.G. (1999, January 20–27). Vision in bad weather. Proceedings of the Seventh IEEE International Conference on Computer Vision, Kerkyra, Greece.
3. He, K., Sun, J., and Tang, X. (2009, January 20–25). Single image haze removal using dark channel prior. Proceedings of the 2009 IEEE Conference on Computer Vision and Pattern Recognition, Miami, FL, USA.
4. A fast single image haze removal algorithm using color attenuation prior;Zhu;IEEE Trans. Image Process.,2015
5. Dehazing using color-lines;Fattal;ACM Trans. Graph.,2014