28M参数高效检测
采用知识蒸馏实现轻量级学生网络学习复杂教师网络的特征提取过程,显著降低计算成本的同时实现轨面缺陷的精确检测。
周武杰 · Yue, Wu · 邱薇薇 · Caie, Xu · Qiang, Fangfang
Applied Soft Computing Journal 2024
参数仅28M,可在移动或嵌入式设备上运行,实现实时检测。
通过知识蒸馏学习复杂教师网络的特征提取过程,保证性能。
双解码结构让双模态特征充分交互,提升对低对比度、暗色调轨面纹理的检测能力。
在NEU RSDDS-AUG工业RGB-D数据集上优于15种先进方法。