投审稿入口

基于注意力机制优化的CAU-Net矿岩图像分割方法研究

Study on CAU-Net Image Segmentation Method Based on Attention Mechanism Optimization

  • 摘要: 针对露天矿爆堆矿岩图像存在的粘连性强、区间物体相似度高以及图像质量差等问题,传统图像分割方法往往分割精度低、效果不佳。为此,在传统U-Net图像分割框架中引入注意力机制,提出了一种基于注意力机制优化的CAU-Net爆堆矿岩图像分割方法。该方法以U-Net为基础架构,在每个卷积学习阶段前融入注意力模块,并与3种注意力模型进行了分割对比试验。结果表明:CAU-Net模型显著提升了分割效果,分割准确度(ACC)达94.08%,AUC (Area Under Curve)值达97.43%,展现出良好的预测性能。进一步与阈值分割法及U-Net法对比,以及交并比指标验证,表明该方法能更精确地分割爆堆矿岩图像,所得高精度分割图像用于数字化矿岩块度统计,大幅提升了统计的准确性,有力推动了矿山数字化建设。

     

    Abstract: To address the challenges of strong adhesion,high inter-object similarity,and poor image quality in open-pit mine blast pile orerock images,conventional segmentation methods often yield low accuracy and unsatisfactory results. To this end,we introduce an attention mechanism into the traditional U-Net framework and propose an attention optimized CAU-Net method for segmenting blast pile ore-rock images. The proposed architecture retains U-Net as the backbone and incorporates attention modules before each convolutional learning stage. Comparative segmentation experiments are conducted against three attention models. The results show that the CAU-Net model significantly improves segmentation performance,achieving an accuracy(ACC) of 94. 08% and an AUC(Area Under Curve) of 97. 43%,indicating strong predictive capability. Further comparisons with threshold-based segmentation and the standard U-Net,together with evaluation using the Intersection over Union(IoU) metric,confirm that the proposed method can segment blast pile ore-rock images more accurately. The high-precision segmentation results,when applied to digital ore-rock fragmentation statistics,substantially enhance statistical accuracy and effectively advance the digitalization of mining operations.

     

/

返回文章
返回