基于改进YOLO模型的隧道围岩裂隙检测与方向预测方法

本发明公开了基于改进YOLO模型的隧道围岩裂隙检测与方向预测方法，包括构建基于YOLOv11架构的裂隙检测模型；将待测图像输入到裂隙检测模型中，裂隙检测模型识别出候选裂隙区域；对候选裂隙区域提取对应的图像块，并将其转换为灰度图像，应用边缘检测器提取边缘特征得到边缘图；通过概率霍夫变换从边缘图中提取几何图元，生成线段候选集合；从线段候选集合中选取最长的线段确定裂隙的主方向，基于主方向进行方向分类。本发明有效提升了对小尺寸和不规则形状裂隙的检测能力，引入基于概率霍夫变换的几何分析模块，在检测的基础上精准估计裂隙的主方向。该端到端框架不仅显著提高了检测精度，还提供了可靠的方向估计，增强了其在实际结构监测中的应用潜力。