摘要
本申请提供了一种模型的训练方法、装置、电子设备、存储介质及程序产品;方法包括:基于样本图像,调用第一目标检测模型和第二目标检测模型分别进行特征提取处理,得到第一图像特征和第二图像特征;根据空间注意力机制、第一图像特征、第二图像特征,确定第一蒸馏损失;根据通道注意力机制、第一图像特征、第二图像特征,确定第二蒸馏损失;将第一蒸馏损失、第二蒸馏损失、第二目标检测模型的目标识别损失的加权求和结果,作为总损失;基于总损失更新第二目标检测模型的参数,得到训练后的第二目标检测模型。通过本申请,能够提升目标检测的准确性。
技术关键词
通道注意力机制
图像
池化特征
计算机可执行指令
蒸馏
模型训练模块
障碍物
电子设备
通道剪枝
计算机程序产品
样本
训练装置
机器人
处理器
参数
可读存储介质
系统为您推荐了相关专利信息
金属表面缺陷
样本生成方法
注意力机制
深度卷积生成对抗网络
融合器
数据检测方法
图像
特征提取模型
斑块
动脉粥样硬化性
AI机器人
探头
检查图像数据
强化学习算法
终端
针织面料
基元
最佳特征
分割方法
多尺度特征提取