Real-time identification of rock chip size and shape distributions from muck images plays a critical role in intelligently optimizing cutterhead thrust and torque parameters for tunnel boring machines(TBM).However,com...Real-time identification of rock chip size and shape distributions from muck images plays a critical role in intelligently optimizing cutterhead thrust and torque parameters for tunnel boring machines(TBM).However,complex light environments in field images are difficult to recognize via traditional methods.This paper proposes a U-Net-SAM framework integrating semantic segmentation and the vision foundation model—Segment Anything Model(SAM),combined with dropout-based uncertainty analysis,achieving efficient rock chip segmentation and parameter quantification.First,a U-Net is trained to identify the rock mass centroid as an automatic SAM prompt.Next,an overlap region optimization strategy based on Intersection over Union(IoU)and a noise filtering method is employed to tackle boundary blurring and particle adhesion.Finally,a Dropout layer is added to implement the committee-based uncertainty analysis model and quantify predictive uncertainty.Results show that:(1)U-Net-SAM improves mean F1-score and PA by 9.1%and 7.8%over U-Net;(2)A strong correlation between prediction standard deviation(SD)and error rate validates the proposed uncertainty quantification strategy.This framework provides reliable rock chip perception for intelligent TBM tunneling,with potential applications in other engineering scenarios.展开更多
基金financial support of National Natural Science Foundation of China(Grant No.52008039)the Natural Science Foundation of Hunan Province(Grant No.2021JJ40592)support from the Research Grants Council of Hong Kong(Grant No.GRF#16208224).
摘要Real-time identification of rock chip size and shape distributions from muck images plays a critical role in intelligently optimizing cutterhead thrust and torque parameters for tunnel boring machines(TBM).However,complex light environments in field images are difficult to recognize via traditional methods.This paper proposes a U-Net-SAM framework integrating semantic segmentation and the vision foundation model—Segment Anything Model(SAM),combined with dropout-based uncertainty analysis,achieving efficient rock chip segmentation and parameter quantification.First,a U-Net is trained to identify the rock mass centroid as an automatic SAM prompt.Next,an overlap region optimization strategy based on Intersection over Union(IoU)and a noise filtering method is employed to tackle boundary blurring and particle adhesion.Finally,a Dropout layer is added to implement the committee-based uncertainty analysis model and quantify predictive uncertainty.Results show that:(1)U-Net-SAM improves mean F1-score and PA by 9.1%and 7.8%over U-Net;(2)A strong correlation between prediction standard deviation(SD)and error rate validates the proposed uncertainty quantification strategy.This framework provides reliable rock chip perception for intelligent TBM tunneling,with potential applications in other engineering scenarios.