一种用于有向目标检测的动态标签分配方法

    公开(公告)号:CN117934810A

    公开(公告)日:2024-04-26

    申请号:CN202311761387.8

    申请日:2023-12-20

    发明人: 魏丹 魏聪聪 江磊

    摘要: 本发明涉及机器学习的技术领域,公开了一种用于有向目标检测的动态标签分配方法,依据特征图上各个有向目标标注框的中心点所在像素单元格的位置信息以及对应有向目标标注框的角度信息,筛选出多个候选样本;构建预测损失计算模型,计算各个特征图中每个候选样本的预测损失;采用OTA标签分配算法中的动态K值估计方法从候选样本中选取正样本参与目标检测网络后续的LOSS计算,以反向传播优化目标检测网络的参数。