煤炭科学技术

北大核心,CA,JST,EI,CSCD

国内刊号:11-2402/TD

国际刊号:0253-2336

煤炭科学技术杂志2024年第1期:基于Transformer和自适应特征融合的矿井低照度图像亮度提升和细节增强方法

发布日期:

作者:田子建, 吴佳奇, 张文琪, 陈伟, 周涛, 杨伟, 王帅,

关键词:图像增强, 图像识别, 生成对抗网络, 特征解耦, Transformer,

高质量矿井影像为矿山安全生产提供保障,也有利于提高后续图像分析技术的性能。矿井影像受低照度环境的影响,易出现亮度低,照度不均,颜色失真,细节信息丢失严重等问题。针对上述问题,提出一种基于Transformer和自适应特征融合的矿井低照度图像亮度提升和细节增强方法。基于生成对抗思想搭建生成对抗式主体模型框架,使用目标图像域而非单一参考图像驱动判别器监督生成器的训练,实现对低照度图像的充分增强;基于特征表示学习理论搭建特征编码器,将图像解耦为亮度分量和反射分量,避免图像增强过程中亮度与颜色特征相互影响从而导致颜色失真问题;设计CEM-Transformer Encoder通过捕获全局上下文关系和提取局部区域特征,能够充分提升整体图像亮度并消除局部区域照度不均;在反射分量增强过程中,使用结合CEM-Cross-Transformer Encoder的跳跃连接将低级特征与深层网络处特征进行自适应融合,能够有效避免细节特征丢失,并在编码网络中添加ECA-Net,提高浅层网络的特征提取效率。制作矿井低照度图像数据集为矿井低照度图像增强任务提供数据资源。试验显示,在矿井低照度图像数据集和公共数据集中,与5种先进的低照度图像增强算法相比,该算法增强图像的质量指标PSNR、SSIM、VIF平均提高了16.564%,10.998%,16.226%和14.438%,10.888%,14.948%,证明该算法能够有效提升整体图像亮度,消除照度不均,避免颜色失真和细节丢失,实现矿井低照度图像增强。

来源:2024年第1期

《煤炭科学技术》期刊编辑部

查看煤炭科学技术杂志2024年第1期

联系我们

  • 地址:北京市和平里青年沟路5号
  • 电话:010-87986431
  • E-mail:mtkxjs@vip.163.com

咨询工作人员