一种基于几何损失函数的轻量化目标检测方法和系统

    公开(公告)号:CN116229147A

    公开(公告)日:2023-06-06

    申请号:CN202310003314.6

    申请日:2023-01-03

    Abstract: 本发明提出了一种基于几何损失函数的轻量化目标检测方法和系统。该方法包括以下步骤:1、从图像识别的数据集中选取多张图片,通过使用标注工具对图像中的物体进行标注并对每张图像进行裁剪操作,将裁剪过后的图像按相应的比例划分为训练集和测试集;2、构建了一种基于几何损失函数的轻量化目标检测器;3、将步骤1中划分好的训练集送入到基于几何损失函数的轻量化目标检测器中进行训练;4、保存训练得到模型,用于之后的目标检测,最终得到图像里面对应目标的类别和位置信息。本发明将图像送入轻量化目标检测器进行训练后,用保存下了的训练模型进行检测,可以达到良好检测效果的同时提高检测速度,并且更适合实际场景的应用。

Patent Agency Ranking