摘要
本发明公开了一种基于知识蒸馏的前视声呐目标检测方法、装置、设备及存储介质,该方法包括:获取前视声呐图像,通过教师模型对前视声呐图像进行特征提取,获得Logits信息;构建轻量化学生模型,并基于Logits信息利用知识蒸馏框架对轻量化学生模型进行蒸馏学习,获得目标学生模型,轻量化学生模型中将注意力机制替换为自适应损失函数;相比于现有技术,本发明通过知识蒸馏框架建立了教师模型与学生模型之间的有效联系,并使得学生模型在保持轻量化的基础上,能够实现高精度的水下目标检测。
技术关键词
前视声呐
蒸馏
学生
焦点损失函数
多尺度特征金字塔
教师
注意力机制
图像
框架
模型训练模块
程序
网络
处理器
检测设备
存储器
物体
参数
数据