一种基于模型轻量化的铁路异物侵限检测方法

本发明公开了一种基于模型轻量化的铁路异物侵限检测方法，包括：S1，获取铁路异物侵限数据集D；S2，构建教师模型并进行训练；S3，构建学生模型，利用铁路异物数据集D对学生模型进行训练，计算学生模型的最终损失函数并更新学生模型的网络参数，直到达到最大迭代次数，得到训练完毕的学生模型，在学生模型的训练过程中，铁路异物数据集D也输入到训练完毕的教师模型；S4，将多个训练完毕的学生模型部署在边缘检测终端，将抽取的视频帧作为铁路异物图像并进行实时检测。本发明的检测方法将教师模型的知识迁移至学生模型，在实现模型轻量化的同时提高了学生模型在铁路异物检测任务中的检测准确率。