一种集成图像上下文特征的钢轨内部伤损识别方法及系统

一种集成图像上下文特征的钢轨内部伤损识别方法及系统，属于铁路轨道探伤技术领域，涉及解决如何提高复杂场景下的钢轨伤损检测精度并增强模型泛化性的问题，本发明使用GAN网络模型对B显图像进行波形实例扩增操作，划分为训练集和测试集；利用不重叠滑窗的窗口Transformer注意力机制构建集成图像上下文特征模块，并用向量平均池化的方法增强模块对图像上下文特征信息的表达能力；通过集成图像上下文特征模块捕获B显图像中的全局噪声杂波点的分布特征和左右轨的出波联系来提高伤损识别模型对波形的识别率，并利用GAN网络模型扩增数据集方式来提高模型的泛化性。