基于改进YOLOv10和SETR的铁轨缺陷检测的分割方法

本发明公开了一种基于改进YOLOv10和SETR的铁轨缺陷检测的分割方法，步骤为：S1：收集图像数据，对图像数据进行增强处理，并进行分割处理，制作数据集；S2：搭建并训练改进的YOLOV10算法模型；S3：使用改进的YOLOV10模型权重对待检测图像数据进行检测，判断是否存在缺陷，若检测出缺陷，则将缺陷的检测框信息进行保存，并输出目标检测框和置信度；S4：将检测框信息作为提示信息传入SETR模型，采用SETR模型对目标进行分割，得到分割结果，输出掩码图。采用改进YOLOv10算法和SETR算法对铁轨缺陷进行检测，既保持了计算复杂度高效检测速度，还提升了检测精度，提高了模型的鲁棒性。