The invention provides a tunneling rock
slag real-time identification method based on U-Net-SAM
coupling driving, and the method achieves the high-precision quantization of the size and morphological parameters of rock
slag through the fusion of a semantic segmentation network (U-net) and a visual basis model (Segment Anything Model, SAM). The method comprises the following steps: firstly, reasoning an input rock
slag image by using a U-Net network to generate a rough rock slag identification result; then, obtaining the center-of-
mass coordinate of each rock
ballast from the rough result as the automatic prompt input of an SAM model, and extracting an initial
mask in combination with the zero sample segmentation capability of the SAM; and carrying out post-
processing optimization on the initial
mask by adopting a morphological
noise filtering method and an intersection-to-union ratio threshold optimization overlapped region
processing strategy. According to the method,
automatic segmentation and size and shape parameter extraction of the tunnel
conveyor belt rock
ballast image are realized, the boundary identification problem of a traditional method in a dense particle scene is effectively solved, and real-time and reliable rock
machine interaction data support is provided for realizing intelligent tunneling parameter regulation and control.