一种目标检测模型训练方法、装置、电子设备及存储介质
基本信息
| 申请号 | CN202210312655.7 | 申请日 | - |
| 公开（公告）号 | CN114663728A | 公开（公告）日 | 2022-06-24 |
| 申请公布号 | CN114663728A | 申请公布日 | 2022-06-24 |
| 分类号 | G06V10/774（2022.01）I;G06K9/62（2022.01）I;G06N3/04（2006.01）I;G06T7/11（2017.01）I;G06T7/62（2017.01）I | 分类 | 计算；推算；计数; |
| 发明人 | 王堃;顾杏春;林业伟;王军 | 申请（专利权）人 | 南京大牛信息科技有限公司 |
| 代理机构 | 北京超凡宏宇专利代理事务所（特殊普通合伙） | 代理人 | - |
| 地址 | 210000江苏省南京市玄武区天山路39号 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本发明的实施例提供了一种目标检测模型训练方法、装置、电子设备及存储介质，方法包括：将待调整图像输入至待训练的目标检测模型中，输出目标预测框；获取目标预测框对应的标准框；确定目标预测框与标准框的交叠率；基于交叠率，计算目标预测框与标准框的目标交叠率；基于目标交叠率，计算目标预测框与标准框之间的损失；将损失输入目标检测模型，以获得更新后的目标检测模型；返回执行确定目标预测框与标准框的交叠率至将损失输入目标检测模型，以获得更新后的目标检测模型的步骤，直至损失收敛为止。通过基于目标预测框和标准框的交叠率，计算目标交叠率，改变现有对目标检测模型的损失函数，从而提升目标检测模型识别的准确率。 |
