一种基于知识蒸馏的铁轨裂纹实时检测方法及系统

本发明公开了一种基于知识蒸馏的铁轨裂纹实时检测方法及系统。具体包括：获取用于进行知识蒸馏训练的铁轨裂纹图像数据集；基于图像集，对教师模型进行训练，获取教师模型部分的网络参数；将教师模型部分的网络参数带入蒸馏网络中，设置多个温度参数，多次进行知识蒸馏训练得到学生模型；将训练后的学生模型从蒸馏网络中提取出来，进行检测精度对比，选取最符合实际需要的模型；将学生模型部署到嵌入式设备中，进行实时检测。本发明针对传统的铁轨裂纹检测精度低且人力资源消耗大的问题，通过知识蒸馏算法在保持模型轻量优势的同时有效提升检测精度，支持在嵌入式设备(如树莓派等)上进行实时的铁轨裂纹检测。