一种目标检测模型训练方法、装置、电子设备及存储介质
基本信息
| 申请号 | CN202110186629.X | 申请日 | - |
| 公开（公告）号 | CN112560999A | 公开（公告）日 | 2021-06-04 |
| 申请公布号 | CN112560999A | 申请公布日 | 2021-06-04 |
| 分类号 | G06K9/62;G06N3/04;G06N3/08 | 分类 | 计算；推算；计数; |
| 发明人 | 王树;罗振波;吉翔 | 申请（专利权）人 | 成都睿沿科技有限公司 |
| 代理机构 | 北京超凡宏宇专利代理事务所（特殊普通合伙） | 代理人 | 蒋姗 |
| 地址 | 610000 四川省成都市高新区中国（四川）自由贸易试验区益州大道中段1800号4栋6层601号 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本申请提供一种目标检测模型训练方法、装置、电子设备及存储介质，该方法包括：获得多个样本图像和多个类别热力图像；以多个样本图像为训练数据，以多个类别热力图像为数据标签，对目标检测模型中的特征提取网络和分类注意力网络进行训练，并以标注图像为数据标签，对目标检测模型中的特征提取网络和检测网络进行训练，通过联合训练的方式获得训练后的目标检测模型。通过在目标检测模型加入了能够提取和有效利用不同类别间的差异化特征的分类注意力网络，从而引导模型去定位待检测图像中的具有重要影响的关键区域，让模型更好地关注关键区域从而学习到易混淆类别之间的差异性，提高了对图像进行目标检测的正确率。 |
