基于Swin-Unet的岩碴图像处理与粒径分布预测

Image processing and particle size distribution prediction of rock chips based on Swin-Unet

  • 摘要: 为解决传统岩碴筛分试验耗时费力,难以为施工提供实时指导,以及现有岩碴图像分割模型数据集内容单一等问题,开展岩碴图像高精度分割模型与岩碴粒径分布预测研究。依托3个典型盾构隧道施工项目,收集隧道掘进段岩碴样品,利用自行搭建的室内拍摄平台获取不同岩性的岩碴图像,构建数据量充足、涵盖多种岩碴类型的图像数据集。基于岩碴图像数据集训练Swin-Unet深度学习网络以实现高精度分割,并与其他主流模型在相同数据集上进行性能对比。将分割后的岩碴图像进行后处理,提取相应的二维图像参数,基于面积法原理预测岩碴粒径分布规律,并将预测结果与岩碴筛分曲线进行对比,验证其合理性。研究结果表明:在相同训练周期下,Swin-Unet模型的ACCDiceIoUHD95分割指标分别为89.51%、85.67%、75.65%、11.91 mm,其表现均优于对比算法,且参数量和计算量相对较小;岩碴粒径分布规律预测结果与岩碴筛分曲线总体吻合,在岩碴粒径小于16 mm区间内预测值略低于筛分值,大于16 mm区间则相反;基于Swin-Unet模型预测结果计算的岩碴粗糙度指数误差在5%以内,预测结果良好。

     

    Abstract: To address the problems that the traditional rock chips screening test is time-consuming and laborious, which is difficult to provide real-time guidance for construction, and the dataset of the existing rock chips image segmentation models is single in content, this study carries out research on high-precision segmentation model of rock chips images and prediction of rock chips particle size distribution. Relying on three typical shield tunnel construction projects, rock chips samples are collected from tunnel boring sections, and rock chips images of different lithologies are obtained by using a self-built indoor shooting platform to construct an image dataset with sufficient data volume and various types of rock chips. The Swin-Unet deep learning network is adopted to train the rock chips image dataset to achieve high-precision segmentation, and its performance is compared with other mainstream models on the same dataset. Post-processing is performed on the segmented rock chips images to extract the corresponding two-dimensional image parameters. The particle size distribution law of rock chips is predicted based on the principle of area method, and the predicted results are compared with the rock chips screening curve to verify its rationality. The results show that under the same training period, the segmentation indexes of ACC, Dice, IoU and HD95 of the Swin-Unet model are 89.51%, 85.67%, 75.65% and 11.91 mm, respectively, which are better than those of the comparison algorithms, with relatively small parameters and computational cost. The predicted results of rock chips particle size distribution are generally consistent with the rock chips screening curve. The predicted values are slightly lower than the screening values when the rock chips particle size is less than 16 mm, and the opposite is true when the particle size is larger than 16 mm. The error of the rock chips roughness index calculated based on the prediction results of the Swin-Unet model is within 5%, indicating a satisfactory prediction effect.

     

/

返回文章
返回