基于迭代压缩U型网络的煤颗粒分割与粒度分析方法

Segmentation and particle size analysis of coal particles based on ISUNet

  • 摘要: 煤中甲烷气体传播与煤粒的粒度分布特征紧密相连,进而影响煤炭的安全开采和利用。随着数字图像处理技术的不断发展,基于数字图像分割的煤粒形态检测方法已成为获取煤颗粒粒度分布特征的主流方法。在数字图像分割过程中,全局信息和边缘细节起着关键作用,直接影响分割结果的准确性。基于卷积神经网络架构的U型网络过于注重局部信息,忽视了全局信息的重要性,容易导致过分割现象。而基于Transformer的网络利用多头自注意力机制有效地建模了全局信息,但却没有充分利用边缘细节特征,导致煤颗粒漏分割问题。为了解决上述问题,本研究提出了迭代压缩U型网络(Iterative Squeeze UNet, ISUNet)用于煤颗粒粒度分析。ISUNet模型引入了压缩激励空洞空间金字塔池化模块和基于Transformer的多路迭代编码器。压缩激励空洞空间金字塔池化模块通过增强不同尺度特征的通道信息和全局上下文信息,解决了煤粒过分割问题。编码器中的多头自注意力模块将ResNet50的卷积特征作为其中一个输入,通过点乘自注意力机制不断强化重要的边缘细节特征,解决了煤粒漏分割问题。与5种经典图像分割模型和4种目前主流的分割模型相比,ISUNet表现出色。相较于经典的分割模型TransUNet来说,平均交并比提高了6.6%,准确率提高了0.3%,召回率提高了7.0%,相较于目前主流的图像分割大模型Segment Anything来说,平均交并比提高了4.6%,准确率提高了0.2%,召回率提高了4.9%。在煤粒粒度测量方面,准确率达到了97.49%。这些试验结果充分证实了ISUNet在煤粒粒度分析中的有效性和优越性。

     

    Abstract: The propagation of methane gas in coal is closely linked to the particle size distribution characteristics of coal particles, which in turn affects the safe mining and utilization of coal. With the continuous development of digital image processing technology, coal particle morphology detection based on digital image segmentation has become the mainstream method for obtaining the particle size distribution characteristics of coal particles. In the process of digital image segmentation, global information and edge details play a crucial role and directly affect the accuracy of the segmentation results. The U-shaped network based on convolutional neural network architecture focus too much on local information, neglecting the importance of global information, which can lead to over-segmentation. On the other hand, Transformer-based networks effectively model global information using multi-head self-attention mechanisms but do not fully utilize edge detail features, resulting in under-segmentation of coal particles. To address these issues, this study proposes an Iterative Squeeze UNet (ISUNet) for coal particle size analysis. The ISUNet model introduces a compressed excitation atrous spatial pyramid pooling module and a Transformer-based multi-path iterative encoder. The compressed excitation atrous spatial pyramid pooling module enhances channel information and global context information of features at different scales, solving the problem of over-segmentation of coal particles. The multi-head self-attention module in the encoder continuously strengthens important edge detail features through dot-product self-attention mechanism, addressing the problem of under-segmentation of coal particles. Compared to five classic image segmentation models and four mainstream segmentation models, ISUNet has shown outstanding performance. Compared to the classic segmentation model TransUNet, it has improved the mean Intersection over Union (mIoU) by 6.6%, the accuracy by 0.3%, and the recall rate by 7.0%. Compared to the current state-of-the-art Segment Anything model, it has improved the mIoU by 4.6%, the accuracy by 0.2%, and the recall rate by 4.9%. In the aspect of coal particle size measurement, the accuracy reached 97.49%. These experimental results fully demonstrate the effectiveness and superiority of ISUNet in coal particle size analysis.

     

/

返回文章
返回