面向列车轮轨接触区域图像分割的生成对抗网络*

列车和轨道之间的开放约束条件决定了车辆脱轨的客观存在。轮轨接触区域边缘曲线分割对列车轮轨接触关系的研究具有重要意义，提出了一种基于生成对抗网络的轮轨接触区域边缘曲线分割算法。通过将残差模块引入生成器网络中，增强了网络对输出变化的敏感程度，进而更好的调整生成器权重。此外，膨胀残差模块的引入，有效扩大了特征图的接收区域。实验结果显示，改进的生成对抗网络对轮轨接触区域边缘曲线的分割准确度达到9613%，敏感度、特异度、F1值、ROC曲线下的面积分别为8390%、9713%、8367%和9812%，验证了该方法能够准确分割轮轨接触区域边缘曲线。