Rw-Dm: Retinex and wavelet-based diffusion model for low-light image enhancement in underground coal mines
摘要
The images in underground coal mines suffer from low contrast and poor visibility. While diffusion model-based approaches for low-light image enhancement have shown considerable promise, these methods are often characterized by high resource consumption, slow processing speeds, and a tendency towards instability. To address these challenges, this paper presents a diffusion model that integrates Retinex and Wavelet transform (RW-DM). This methodology harnesses the generative capabilities of diffusion models to produce visually appealing outcomes in the enhancement of low-light images. This approach decomposes low-light images into reflectance and illumination components, utilizing the advantages of wavelet transformation to significantly enhance processing speed and reduce computational resource consumption while preserving data integrity. Additionally, the study introduces a Multiple Residual Module (MRM) to preserve details in both the decomposing network (MR-Decompose) and the denoising network (MRU-Denoise). Extensive experiments on low-light images from coal mines demonstrate that RW-DM outperforms existing state-of-the-art methods qualitatively. Moreover, the application of the proposed method in pedestrian detection using underground coal mine images resulted in a 3.1% increase in