一种基于知识蒸馏的无人机影像云边协同识别方法
摘要:
本发明公开了一种基于知识蒸馏的无人机影像云边协同识别方法,解决了现有技术的不足,包括步骤1,构建老师模型,并对老师模型进行训练是老师模型完全收敛并具备指导能力;步骤2,构建学生模型,改进学生模型的损失函数使学生模型能同时接受来自人工标签和老师模型的软标签;步骤3,对老师模型和学生模型进行联合训练,学生模型同时接受人工标签和老师模型的软标签进行迭代,直至学生模型完全收敛;步骤4,将训练完成的学生模型部署至无人机终端,使学生模型在边缘端实际运行并检测识别效果;步骤5,若检测识别效果通过判断为不达标,则将识别的图像传输至老师模型,老师模型通过云边协同的方式指导学生模型进行二次训练至识别效果达标为止。
0/0