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

Underexposed image enhancement method for high-belt-speed coal-gangue sorting conditions

  • 摘要: 针对高带速煤矸分拣工况下因曝光时间受限引发的图像欠曝问题,提出一种基于HVI颜色空间的欠曝图像增强网络HVI−MTDB−Net。将欠曝煤矸图像由RGB颜色空间映射至HVI颜色空间,实现光照与色彩的有效解耦;构建多任务双分支网络,通过共享编码器提取多尺度特征,并在编码阶段设计递归上下文聚合器(RCA)以提升特征提取能力;解码阶段设计光照增强分支I−Net和色彩恢复分支HV−Net,分别完成光照增强与色彩恢复。基于高带速成像曝光时间约束及图像特征分析,确定不同带速下差异化曝光策略,并据此构建200组静态配对数据集与2 400张实时高带速欠曝数据集,分别用于网络训练与泛化能力测试。试验结果表明:HVI颜色空间对高带速工况下煤矸图像的欠曝退化特征具有较好的适配能力,客观评价指标均优于基于YUV,Lab和HSV颜色空间的结果;HVI−MTDB−Net在测试集上的PSNR,SSIM,EN和GM分别达到21.875,0.842,7.931和4.786;以HVI−MTDB−Net增强后的图像作为YOLOv11s输入时,mAP@0.5:0.95由原始欠曝条件下的0.061提升至0.551,相对次优对比方法提高6.6%,网络推理帧率达到156 帧/s,满足高带速工况下的检测精度和实时性需求。

     

    Abstract: Limited exposure time in high-belt-speed coal-gangue sorting causes underexposed images. To address this problem, a Multi-Task Dual-Branch Network for Underexposed Image Enhancement, HVI-MTDB-Net, was proposed based on the Horizontal/Vertical-Intensity (HVI) color space. Underexposed coal-gangue images were mapped from the standard Red-Green-Blue (sRGB) color space to the HVI color space to effectively decouple illumination and color. A multi-task dual-branch network was constructed, in which a shared encoder was used to extract multiscale features and a recurrent context aggregator was designed in the encoding stage to improve feature extraction capability. During decoding and enhancement, an Illumination Enhancement Network (I-Net) and a Horizontal-Vertical Color Restoration Network (HV-Net) were designed to perform illumination enhancement and color restoration, respectively. Based on the exposure-time constraints of high-belt-speed imaging and image feature analysis, speed-specific exposure strategies were determined, and 200 groups of static paired dataset and a real-time high-belt-speed underexposed image dataset containing 2 400 images were constructed for network training and generalization testing, respectively. Experimental results showed that HVI-MTDB-Net achieved PSNR, SSIM, EN, and GM values of 21.875, 0.842, 7.931, and 4.786, respectively, on the test set, outperforming 10 mainstream image-enhancement methods, including RetinexNet and EnlightenGAN. When images enhanced by HVI-MTDB-Net were used as input to YOLOv11s, mAP@0.5:0.95 increased from 0.061 under the original underexposed condition to 0.551, and the frame rate reached 156 frames/s, meeting the requirements for detection accuracy and real-time performance under high-belt-speed conditions.

     

/

返回文章
返回