Citation: | WANG Yiwei, LI Xiaoyu, WENG Zhi, et al. Low-light image enhancement method for underground mines based on an improved Zero-DCE model[J]. Journal of Mine Automation,2025,51(2):57-64, 99. DOI: 10.13272/j.issn.1671-251x.2024110072 |
Underground coal mine surveillance images suffer from noise, low clarity, missing color, and texture information. Additionally, machine learning-based image enhancement methods face challenges in collecting paired low-light and normal-light image datasets. To address these issues, this paper proposes an improved Zero-Reference Deep Curve Estimation (Zero-DCE) model for enhancing low-light images in mines. The ReLU activation function in the Zero-DCE model was replaced with Leaky ReLU to accelerate model convergence and improve the efficiency of low-light image feature learning. A Convolutional Block Attention Module (CBAM) was introduced at the skip connections between the shallow and deep networks of the Zero-DCE model to enhance the model's ability to capture key image features. An Asymmetric Convolution Block (ACB) was incorporated into the shallow network to optimize the model's learning of local image features and its ability to represent fine details. A Cascaded Convolution Kernel (CCK) was employed in the deep network to reduce the number of model parameters and computational cost, thereby shortening the training time. Experimental validation was conducted using the LOL public dataset and a self-built mine dataset. The results showed that the improved Zero-DCE model outperformed typical image enhancement models in terms of Mean Squared Error (MSE), Peak Signal-to-Noise Ratio (PSNR), Structural Similarity (SSIM), Natural Image Quality Evaluator (NIQE), and Visual Information Fidelity (VIF). Specifically, on the self-built dataset, MSE and NIQE decreased by 16.25% and 2.93%, respectively, while PSNR, SSIM, and VIF increased by 2.87%, 1.87%, and 17.64%, respectively. The enhanced images exhibited superior visual quality, effectively improving brightness while preserving detailed texture information and significantly reducing noise. The inference time for a single image was 0.138 seconds, enabling real-time image enhancement.
[1] |
程德强,钱建生,郭星歌,等. 煤矿安全生产视频AI识别关键技术研究综述[J]. 煤炭科学技术,2023,51(2):349-365.
CHENG Deqiang,QIAN Jiansheng,GUO Xingge,et al. Review on key technologies of AI recognition for videos in coal mine[J]. Coal Science and Technology,2023,51(2):349-365.
|
[2] |
范伟强,刘毅. 基于自适应小波变换的煤矿降质图像模糊增强算法[J]. 煤炭学报,2020,45(12):4248-4260.
FAN Weiqiang,LIU Yi. Fuzzy enhancement algorithm of coal mine degradation image based on adaptive wavelet transform[J]. Journal of China Coal Society,2020,45(12):4248-4260.
|
[3] |
吴佳奇,张文琪,陈伟,等. 基于改进CycleGAN的煤矿井下低照度图像增强方法[J]. 华中科技大学学报(自然科学版),2023,51(5):40-46.
WU Jiaqi,ZHANG Wenqi,CHEN Wei,et al. Image enhancement method of underground low illumination in coal mine based on improved CycleGAN[J]. Journal of Huazhong University of Science and Technology(Natural Science Edition),2023,51(5):40-46.
|
[4] |
程健,李昊,马昆,等. 矿井视觉计算体系架构与关键技术[J]. 煤炭科学技术,2023,51(9):202-218.
CHENG Jian,LI Hao,MA Kun,et al. Architecture and key technologies of coalmine underground vision computing[J]. Coal Science and Technology,2023,51(9):202-218.
|
[5] |
田会娟,蔡敏鹏,关涛,等. 基于YCbCr颜色空间的Retinex低照度图像增强方法研究[J]. 光子学报,2020,49(2):173-184.
TIAN Huijuan,CAI Minpeng,GUAN Tao,et al. Low-light image enhancement method using Retinex method based on YCbCr color space[J]. Acta Photonica Sinica,2020,49(2):173-184.
|
[6] |
GUO Xiaojie,LI Yu,LING Haibin. LIME:low-light image enhancement via illumination map estimation[J]. IEEE Transactions on Image Processing :a Publication of the IEEE Signal Processing Society,2016,26(2):982-993.
|
[7] |
WEI Chen,WANG Wenjing,YANG Wenhan,et al. Deep Retinex decomposition for low-light enhancement[Z/OL]. [2024-10-11]. https://doi.org/10.48550/arXiv.1808.04560.
|
[8] |
ZHANG Yonghua,ZHANG Jiawan,GUO Xiaojie. Kindling the darkness:a practical low-light image enhancer[C]. The 27th ACM International Conference,Torino, 2019. DOI: 10.1145/3343031.3350926.
|
[9] |
ZHANG Yonghua,GUO Xiaojie,MA Jiayi,et al. Beyond brightening low-light images[J]. International Journal of Computer Vision,2021,129(4):1-25.
|
[10] |
王满利,张航,李佳悦,等. 基于深度神经网络的煤矿井下低光照图像增强算法[J]. 煤炭科学技术,2023,51(9):231-241.
WANG Manli,ZHANG Hang,LI Jiayue,et al. Deep neural network-based image enhancement algorithm for low-illumination images underground coal mines[J]. Coal Science and Technology,2023,51(9):231-241.
|
[11] |
GUO Chunle,LI Chongyi,GUO Jichang,et al. Zero-reference deep curve estimation for low-light image enhancement[C]. The IEEE/CVF Conference on Computer Vision and Pattern Recognition,Seattle,2020:1777-1786.
|
[12] |
JIANG Yifan,GONG Xinyu,LIU Ding,et al. EnlightenGAN:deep light enhancement without paired supervision[J]. IEEE Transactions on Image Processing:a Publication of the IEEE Signal Processing Society,2021,30:2340-2349. DOI: 10.1109/TIP.2021.3051462
|
[13] |
ZHANG Yu,DI Xiaoguang,ZHANG Bin,et al. Self-supervised image enhancement network:training with low light images only[Z/OL]. [2024-10-11]. https://doi.org/10.48550/arXiv.2002.11300.
|
[14] |
田子建,阳康,吴佳奇,等. 基于LMIENet图像增强的矿井下低光环境目标检测方法[J]. 煤炭科学技术,2024,52(5):222-235.
TIAN Zijian,YANG Kang,WU Jiaqi,et al. LMIENet enhanced object detection method for low light environment in underground mines[J]. Coal Science and Technology,2024,52(5):222-235.
|
[15] |
ZHANG Jian,CRAIGMILE P F,CRESSIE N. Loss function approaches to predict a spatial quantile and its exceedance region[J]. Technometrics,2008,50(2):216-227. DOI: 10.1198/004017008000000226
|
[16] |
YASHWANT K,VIJAYSHRI C. An image fusion approach based on adaptive fuzzy logic model with local level processing[J]. International Journal of Computer Applications,2015,124(1):39-42. DOI: 10.5120/ijca2015905317
|
[17] |
BERLOFFA G,MODENA F. Income shocks,coping strategies,and consumption smoothing:an application to Indonesian data[J]. Journal of Asian Economics,2013,24:158-171. DOI: 10.1016/j.asieco.2012.11.004
|
[18] |
LI Zhengguo,ZHENG Jinghong,RAHARDJA S. Detail-enhanced exposure fusion[J]. IEEE Transactions on Image Processing,2012,21(11):4672-4676. DOI: 10.1109/TIP.2012.2207396
|
[19] |
XU Bing,WANG Naiyan,CHEN Tianqi,et al. Empirical evaluation of rectified activations in convolutional network[Z/OL]. [2024-10-11]. https://doi.org/10.48550/arXiv.1505.00853.
|
[20] |
DING Xiaohan, GUO Yuchen, DING Guiguang, et al. ACNet: strengthening the kernel skeletons for powerful CNN via asymmetric convolution blocks[J]. International Conference on Computer Vision, Seoul, 2019. DOI: 10.1109/ICCV.2019.00200.
|
[21] |
WOO S,PARK J,LEE J Y,et al. CBAM:convolutional block attention module[C]. European Conference on Computer Vision,Munich,2018:3-19.
|
[22] |
苗作华,赵成诚,朱良建,等. 矿井井下非均匀照度图像增强算法[J]. 工矿自动化,2023,49(11):92-99.
MIAO Zuohua,ZHAO Chengcheng,ZHU Liangjian,et al. Image enhancement algorithm for non-uniform illumination in underground mines[J]. Journal of Mine Automation,2023,49(11):92-99.
|
[23] |
赵征鹏,李俊钢,普园媛. 基于卷积神经网络的Retinex低照度图像增强[J]. 计算机科学,2022,49(6):199-209.
ZHAO Zhengpeng,LI Jungang,PU Yuanyuan. Low-light image enhancement based on Retinex theory by convolutional neural network[J]. Computer Science,2022,49(6):199-209.
|
[24] |
SETIADI D R I M. PSNR vs SSIM:imperceptibility quality assessment for image steganography[J]. Multimedia Tools and Applications,2020,80(6):1-22.
|
1. |
张帆,王广森,潘仁杰,宋雨祥. 定向复合钻进预抽煤层中部瓦斯的以孔代巷技术研究. 煤. 2023(02): 33-36 .
![]() | |
2. |
张帆,王晨辉,杨镜渝,王彬彬,李子江. 大采长工作面中部煤层预抽瓦斯以孔代巷技术应用. 煤炭技术. 2023(05): 121-125 .
![]() |