钢轨扣件弹条的缺陷图像生成方法及检测方法

本发明涉及轨道交通技术领域，具体涉及钢轨扣件弹条的缺陷图像生成方法及检测方法，包括以下步骤：S1，获取扣件图像：获取真实场景下的扣件图像；S2，标记扣件掩码图像：标记正常扣件图像为扣件掩码图像；S3，生成缺陷扣件掩码图像：生成不同类型的弹条缺陷的扣件掩码图像；S4，风格迁移生成缺陷图像：构建钢轨扣件弹条缺陷图像数据集；S5，构建训练数据集：构建训练数据集；S6，构建缺陷检测模型：通过训练数据集对弹条缺陷检测模型进行训练；S7，缺陷检测：对待检测扣件图像进行缺陷检测。本发明，提升了钢轨扣件弹条缺陷检测的精度、效率以及鲁棒性，能够在轨道交通中大幅提高扣件缺陷的自动化检测能力。