An illuminance improvement and details enhancement method on coal mine low-light images based on Transformer and adaptive feature fusion

被引:0
|
作者
Tian Z. [1 ]
Wu J. [1 ]
Zhang W. [1 ]
Chen W. [1 ,2 ]
Zhao T. [3 ]
Yang W. [1 ]
Wang S. [1 ,4 ]
机构
[1] School of Mechanical Electronic & Information Engineering, China University of Mining and Technology- Beijing, Beijing
[2] School of Computer Science & Technology, China University of Mining and Technology, Xuzhou
[3] School of Petroleum Engineering, China University of Petroleum (East China), Qingdao
[4] Inner Mongolia Bureau, the National Mine Safety Administration, Inner Mongolia Autonomous Region, Hohhot
关键词
feature disentanglement; generative adversarial network; image enhancement; image recognition; Transformer;
D O I
10.13199/j.cnki.cst.2023-0112
中图分类号
学科分类号
摘要
High quality mine images can provide guarantee for mine safety production, and improve the performance of subsequent image analysis technologies. Affected by low illuminance environment, mine images suffer low brightness, uneven brightness, color distortion, and serious loss of details. Aiming at the above problems, an illuminance improvement and details enhancement method on coal mine low-light images based on Transformer and adaptive feature fusion was proposed to enhance the brightness and detail of mine low illuminance images. Based on the idea of generative confrontation, a framework of generative adversary agent model was built, and the target image domain was used instead of a single reference image to drive discriminator to supervise the training of generator, so as to achieve adaptive enhancement of low illuminance images; The feature encoder was built based on the feature representation learning theory to decouple the image into illuminance component and reflection component, the method can avoid the interaction between illuminance and color features during image enhancement to solve the color distortion; the CEM-Transformer Encoder was designed to enhance the brightness component, the method can improve the overall image brightness and eliminate the local area brightness unevenness, by capturing the global context and extracting the local area features; In the process of reflection component enhancement, the skip connection combined with CEM-Cross-Transformer Encoder was used to adaptively fuse low-level features with features at the deep CNN layers, which can effectively avoid the loss of detailed features, and ECA-Net was added to the encode network to improve the feature extraction efficiency of the shallow CNN layers. The low illuminance mine image dataset was produced to provide data resources for the low illuminance mine image enhancement task. The experiments show that, compared with five advanced low illuminance image enhancement algorithms, the quality indicators PSNR, SSIM and VIF of the images enhanced by the algorithm are improved by 16.564%, 10.998%, 16.226% and 14.438%, 10.888% and 14.948% on average on the low illuminance mine image dataset and the public dataset. And the algorithm also perform well in subjective visual evaluation. The above results prove that the algorithm can effectively improve the overall image brightness and eliminate the uneven brightness, thus to achieve mine low illuminace image enhancement. © 2024 China Coal Society. All rights reserved.
引用
收藏
页码:297 / 310
页数:13
相关论文
共 40 条
  • [1] WANG Guofa, WANG Hong, REN Huaiwei, Et al., 2025 scenarios and development path of intelligent coal mine[J], Journal of China Coal Society, 43, 2, pp. 295-305, (2018)
  • [2] Wei CHEN, REN Peng, TIAN Zijian, Et al., Unsupervised mine personnel tracking based on attention mechanism[J], Journal of China Coal Society, 46, S1, pp. 601-608, (2021)
  • [3] HAN Jianghong, Xing WEI, LU Yang, Et al., Driverless technology of underground locomotive in coal mine[J], Journal of China Coal Society, 45, 6, pp. 2104-2115, (2020)
  • [4] YANG Xiao, Wei CHEN, REN Peng, Et al., Coal mine monitoring image semantic segmentation based on domain adaptation[J], Journal of China Coal Society, 46, 10, pp. 3386-3396, (2021)
  • [5] HAO Shuai, ZHANG Xu, MA Xu, Et al., Foreign object detection in coal mine conveyor belt based on CBAM-YOLOv5[J/OL], Journal of China Coal Society
  • [6] SHAN Pengfei, SUN Haoqiang, LAI Xingping, Et al., Identification method on mixed and release state of coal-gangue masses of fully mechanized caving based on improved Faster R-CNN[J], Journal of China Coal Society, 47, 3, (2022)
  • [7] 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, 51, 2, pp. 349-365, (2023)
  • [8] Yeong-Taeg KIM, Contrast enhancement using brightness preserving bi-histogram equalization[J], IEEE transactions on Consumer Electronics, 43, 1, (1997)
  • [9] Yu WANG, Qian CHEN, Baeomin ZHANG, Image enhancement based on equal area dualistic sub-image histogram equalization method[J], IEEE transactions on Consumer Electronics, 45, 1, (1999)
  • [10] Jianwei WANG, An enhancement algorithm for low-illumination color image with preserving edge[J], Computer Technology Development, 28, 1, (2018)