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