面向高带速煤矸分拣工况的欠曝图像增强方法研究

Research on Underexposed Image Enhancement Method for High-Speed Coal Gangue Sorting Working Conditions

  • 摘要: 针对高带速条件下,煤矸图像因曝光受限引发光照不足、细节模糊缺失及色彩失真等退化问题。本文提出一种基于多任务双分支架构的高带速欠曝煤矸图像增强网络HVI-MTDB-Net(HVI-based Multi-Task Dual-Branch Network for Underexposed Image Enhancement)。该网络先将原始欠曝煤矸RGB图像转换至HVI颜色空间;结合其光照-色彩独立特性,设计多任务双分支架构,通过共享编码器提取多尺度特征,并在编码器中集成递归上下文聚合器(RCA)以提升特征表达精准度;通过光照增强分支(I-Net)与色彩恢复分支(HV-Net)实现强度分支与水平/垂直分支的协同引导交互,I-Net增加光照引导模块(IPG)实现暗区优先增强与边缘保护,HV-Net增加色彩融合模块(CFM)实现细节保留与色彩自然还原的同步提升。在实际煤矸采集环境下构建的高带速欠曝图像数据集上的实验结果表明,HVI-MTDB-Net在PSNR、SSIM、EN、GM等图像质量指标上均优于YUV、Lab、HSV等传统颜色空间及RetinexNet、EnlightenGAN等主流方法。在煤矸识别任务中,经HVI-MTDB-Net增强的图像显著提升了YOLOv11n模型的准确率、召回率及mAP,其中mAP@0.5:0.95相较次优方法提高6.6%,验证了该方法在工业欠曝视觉增强场景中的有效性与工程应用价值。

     

    Abstract: Under high-speed conditions, coal gangue images suffer from degradation issues such as insufficient illumination, blurred details, and color distortion due to exposure limitations. This paper proposes a high-speed underexposed coal gangue image enhancement network based on a multi-task dual-branch architecture, termed HVI-MTDB-Net (HVI-based Multi-Task Dual-Branch Network for Underexposed Image Enhancement). The network first converts the original underexposed coal gangue RGB images to the HVI color space. Exploiting the light-color independence property of the HVI space, a multi-task dual-branch architecture is designed. This architecture utilizes a shared encoder to extract multi-scale features, integrating a Recursive Context Aggregator (RCA) within the encoder to enhance feature representation accuracy. The network achieves collaborative guidance interaction between the intensity branch (I-Net) and the color recovery branch (HV-Net). The I-Net introduces an illumination guidance module (IPG) to prioritize enhancement in dark regions while protecting edges, and the HV-Net incorporates a color fusion module (CFM) to simultaneously improve detail preservation and natural color restoration.Experimental results on a high-speed underexposed image dataset constructed in a real coal gangue collection environment demonstrate that HVI-MTDB-Net outperforms traditional color spaces such as YUV, Lab, and HSV, as well as mainstream methods like RetinexNet and EnlightenGAN, in image quality metrics including PSNR, SSIM, EN, and GM. In coal gangue recognition tasks, images enhanced by HVI-MTDB-Net significantly improve the accuracy, recall, and mAP of the YOLOv11n model. Notably, the mAP@0.5:0.95 increases by 6.6% compared to the next best method, validating the effectiveness and industrial application potential of the proposed method in underexposed visual enhancement scenarios.

     

/

返回文章
返回