基于网络蒸馏模型的供电接触网支撑装置零部件快速定位方法

为确保供电接触网支撑装置具备良好的工作状态，避免支撑装置失效可能造成的严重弓网故障，进而影响列车运行安全，提出一种新型的、适用于小型化嵌入式设备进行部署应用的、基于网络蒸馏技术的深度学习接触网支撑装置快速定位识别模型。网络蒸馏模型是基于双深度学习网络进行指导性学习的，其利用高精度深度学习模型YOLOv5x、YOLOv5l、YOLOv5m作为蒸馏算法中的教师模型，然后利用网络蒸馏技术对复杂模型的特征提取能力进行迁移，以教师模型的类别概率作为软目标并将其用于训练学生模型，从而将知识从复杂模型(教师模型)转移到更高效的小型化模型(学生模型)。研究结果表明，利用网络蒸馏技术可以获得精度高、速度快、模型小、易于部署的接触网支撑装置定位模型。