密集堆叠岩屑图像颗粒细粒度分割方法

Fine-Grained Particle Segmentation Method for Dense Stacked Rock Debris Images

  • 摘要: 为解决岩屑分布密集图像边界存在重叠导致图像分割效率低下、小目标漏检和边界分割精度较差等问题,研制了岩屑图像颗粒细粒度分割网络。该网络基于Mask R-CNN网络架构,采用DW轻量卷积构建主干网络,利用坐标注意力机制构建了多尺度特征金字塔,以增强特征选择性和空间位置信息的捕捉能力,并提升不同尺寸目标的检测能力;结合基于距离图引导的边界增强损失函数,进一步优化网络对细粒度边界特征的学习,提升分割结果的精确度。采用丰谷、新场等区块采集的岩屑图像样本对改进后的岩屑分割网络进行测试,平均精度为92.1%,平均交并比为83.2%。研究表明,改进后的岩屑图像颗粒细粒度分割网络的推理速度及精度较现有分割网络明显提升,为自动化岩屑图像分析提供了技术支持。

     

    Abstract: To solve the problems of low image segmentation efficiency, missed detection of small targets, and poor boundary segmentation accuracy caused by overlapping boundaries in images with dense rock debris distribution, a fine-grained segmentation network for rock debris images was developed. This model was based on the Mask R-CNN network architecture. It used DW lightweight convolution to build the backbone network and utilized coordinate attention mechanism to construct a multi-scale feature pyramid to enhance feature selectivity and capture spatial position information, so as to improve the detection ability of targets of different sizes. By combining the distance graph-guided boundary enhancement loss function, the model’s learning of fine-grained boundary features was further optimized, and the accuracy of segmentation results was improved. The improved rock debris segmentation network was tested using rock debris image samples collected from blocks such as Fenggu and Xinchang. The average precision was 92.1%, and the mean intersection over union(mIOU) was 83.2%. The results show that the improved rock debris segmentation network has significantly improved inference speed and accuracy compared to existing segmentation networks, providing technical support for automated rock debris image analysis.

     

/

返回文章
返回