摘要
本申请属于目标检测技术领域,尤其涉及一种目标检测模型训练方法、装置、计算机可读存储介质及电子设备。所述方法包括:对预设的目标检测教师模型进行模型压缩,得到与目标检测教师模型对应的目标检测学生模型;生成与输入图像对应的二值掩码;通过二值掩码分别对第一特征图和第二特征图进行特征掩码过滤,得到第一掩码特征图和第二掩码特征图;其中,第一特征图为输入图像经由目标检测教师模型提取得到的特征图,第二特征图为输入图像经由目标检测学生模型提取得到的特征图;根据第一掩码特征图和第二掩码特征图,确定目标检测学生模型的损失函数;基于损失函数对目标检测学生模型进行训练,得到训练后的目标检测模型。
技术关键词
检测模型训练方法
检测教师
学生
模型压缩
像素点
图像
可读存储介质
蒸馏
模型训练模块
电子设备
处理器
偏差
数值
元素
过滤模块
计算机
存储器
系统为您推荐了相关专利信息
特征分析系统
动态可视化
分析模块
纹路特征
多参数