一种标签四点检测模型构建方法和装置

    公开(公告)号:CN116993969A

    公开(公告)日:2023-11-03

    申请号:CN202310958596.5

    申请日:2023-07-31

    Abstract: 本发明公开了一种标签四点检测模型构建方法和装置,所述方法包括:获取标签图像的训练数据集,对其进行数据增强并标注正确的目标检测框;构建初始网络模型,该模型包括主干网络、位置子网络和分类子网络;通过主干网络提取并生成相应图像样本的多层特征图;再通过分类子网络和位置子网络生成每个目标的初始类别和位置;将多层特征图中每个目标的预测框映射回原图,计算得到多个经过还原的真实坐标的预测框,采用预设方法筛选得到最优的预测框的类别及位置;采用训练数据集训练初始网络模型,构建损失函数,以得到标签四点检测模型。本发明能够有效检测不规则四边形目标,精确输出不规则四边形的四个角点坐标。

Patent Agency Ranking