一种基于DETR的端到端接触网部件缺陷检测方法

本发明提供了一种基于DETR的端到端接触网部件缺陷检测方法，涉及铁路基础设施设备检测技术领域，包括Backbone模块、Encoder模块和Decoder模块，所述Backbone模块、Encoder模块和Decoder模块均基于DETR检测模型，所述Backbone模块用于提取采集的图像中的纹理和语义特征，将输入的二维图像转化为特征向量；所述Encoder模块为网络的编码器，用于将Backbone模块提取的图像特征编码成解码所需的状态向量；本发明使用基于DETR的端到端目标检测方法，增强模型的图像理解和目标识别能力，并针对小目标检测进行改进，从而更好地实现接触网缺陷自动化检测的目标。