公开(公告)号：CN114677576A

公开(公告)日：2022-06-28

申请号：CN202210253989.1

申请日：2022-03-15

Applicant: 南京邮电大学

Inventor： 邵文泽 ,  沈翔

IPC: G06V20/00 ,  G06V10/774 ,  G06V10/764 ,  G06V10/82 ,  G06K9/62 ,  G06N3/04 ,  G06N3/08

Abstract: 本发明公开了一种用于目标检测的自适应平衡训练损失的方法、装置及存储介质，方法包括：将训练样本输入检测模型，得到输出的用于计算损失的数据，包括预测框的位置、预测框的宽高以及预测框分类的得分；将损失平衡因子用于分类损失，对检测模型进行训练，优化计算得到分类损失平衡因子；将损失平衡因子用于边框回归损失，对检测模型进行训练，优化计算得到边框回归损失平衡因子；将所述分类损失平衡因子和边框回归损失平衡因子同时用于训练总损失，对检测模型进行训练，优化得到最优检测模型。实验结果证明，本发明设计出的损失平衡因子能有效平衡目标损失，提高检测器的平均精度，且数据集中的各个种类的检测精度都得到了不同程度的提升。

公开(公告)号：CN114677576B

公开(公告)日：2024-10-29

申请号：CN202210253989.1

申请日：2022-03-15

Applicant: 南京邮电大学

Inventor： 邵文泽 ,  沈翔

IPC: G06V20/00 ,  G06V10/774 ,  G06V10/764 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/08

Abstract: 本发明公开了一种用于目标检测的自适应平衡训练损失的方法、装置及存储介质，方法包括：将训练样本输入检测模型，得到输出的用于计算损失的数据，包括预测框的位置、预测框的宽高以及预测框分类的得分；将损失平衡因子用于分类损失，对检测模型进行训练，优化计算得到分类损失平衡因子；将损失平衡因子用于边框回归损失，对检测模型进行训练，优化计算得到边框回归损失平衡因子；将所述分类损失平衡因子和边框回归损失平衡因子同时用于训练总损失，对检测模型进行训练，优化得到最优检测模型。实验结果证明，本发明设计出的损失平衡因子能有效平衡目标损失，提高检测器的平均精度，且数据集中的各个种类的检测精度都得到了不同程度的提升。