【技术实现步骤摘要】
评估目标检测模型的方法、获取测试集的方法和装置
[0001]本申请涉及计算机
，特别是涉及一种评估轨迹跟踪系统中目标检测模型的方法和装置，以及一种获取测试集的方法和装置。
技术介绍
[0002]目标检测模型，能够检测出图片中的目标(如：人、车等)。轨迹跟踪系统借助目标检测模型以及轨迹生成逻辑，以获得视频中目标在空间的移动轨迹。轨迹跟踪系统中的目标检测模型的精度对于轨迹跟踪系统的影响至关重要，目前，评价目标检测模型的参数，通常是以目标检测模型所检测的目标的检测框为粒度的，体现的是目标检测模型好坏，而不能反映目标检测模型是否达到了轨迹跟踪系统的使用要求。
[0003]基于此，亟待提供一种能够反映目标检测模型在轨迹跟踪系统中工作情况的技术方案，用于评估轨迹跟踪系统中目标检测模型的好坏，使得优质的轨迹跟踪系统被用户使用且为用户提供较好的服务。
技术实现思路
[0004]本申请实施例提供了一种评估轨迹跟踪系统中目标检测模型的方法和装置，通过新定义的指标，能够合理的评估轨迹跟踪系统中目标检测模型的工作情况，...
【技术保护点】
【技术特征摘要】
1.一种评估轨迹跟踪系统中目标检测模型的方法，其特征在于，包括：基于测试集中的多张图片和轨迹跟踪系统的目标检测模型，获得测试结果，所述测试集包括所述多张图片的标注结果，所述测试结果和所述标注结果均包括图片在视频帧中的编号和检测框位置信息，所述标注结果还包括轨迹标识；根据所述测试结果和所述标注结果，获得所述目标检测模型的轨迹评估指标的值，所述轨迹评估指标用于体现所述目标检测模型在所述轨迹跟踪系统中的工作情况。2.根据权利要求1所述的方法，其特征在于，所述轨迹评估指标包括轨迹召回率和/或轨迹断开率，所述轨迹召回率取召回轨迹数量和总轨迹数量之比，所述轨迹断开率取所有轨迹的断开总次数和召回轨迹数量之比，目标连续N帧被检测到则该目标记作一个召回轨迹，所述N为大于或等于1的整数。3.根据权利要求1或2所述的方法，所述根据所述测试结果和所述标注结果，计算所述目标检测模型的轨迹评估指标的值，包括：根据所述测试结果和所述标注结果获得评估结果，所述评估结果包括：每个轨迹标识、该轨迹出现在所述测试结果中的编号以及该轨迹出现在所述标注结果中的编号之间的对应关系，所述评估结果用于表征各轨迹在所述测试结果和所述标注结果中的出现情况；根据所述评估结果计算所述目标检测模型的轨迹评估指标的值。4.根据权利要求3所述的方法，其特征在于，所述测试集中包括第一图片，所述测试结果中包括所述第一图片的第一测试框，所述标注结果中包括所述第一图片的第一标注框，针对所述第一图片，所述根据所述测试结果和所述标注结果获得评估结果，包括：计算所述第一测试框与所述第一标注框的第一重叠度IoU；确定所述第一IoU选择满足预设条件，将所述第一测试框的轨迹标识设置为所述第一标注框的第一轨迹标识，所述预设条件包括：IoU中的最大值大于第二阈值。5.根据权利要求3所述的方法，其特征在于，所述测试集中包括第一图片，所述测试结果中包括所述第一图片的第一测试框和第二测试框，所述标注结果中包括所述第一图片的第一标注框和第二标注框，针对所述第一图片，所述根据所述测试结果和所述标注结果获得评估结果，包括：计算所述第一测试框与所述第一标注框的第一重叠度IoU、所述第一测试框与所述第二标注框的第二IoU、所述第二测试框与所述第一标注框的第三IoU以及所述第二测试框与所述第二标注框的第四IoU；从所述第一IoU和所述第二IoU中选择满足预设条件的所述第一IoU，将所述第一测试框的轨迹标识设置为所述第一标注框的第一轨迹标识；并且，从所述第三IoU和所述第四IoU中选择满足所述预设条件的所述第四IoU，将所述第二测试框的轨迹标识设置为所述第二标注框的第二轨迹标识，所述评估结果包括所述第一测试框的所述第一轨迹标识和所述第二测试框的所述第二轨迹标识，所述预设条件包括：IoU中的最大值且所述最大值大于第二阈值。6.根据权利要求5所述的方法，其特征在于，所述第一图片还包括第三测试框，针对所述第一图片，所述根据所述测试结果和所述标注结果，获得评估结果，还包括：计算所述第三测试框与所述第一标注框的第五IoU以及所述第三测试框与所述第二标注框的第六IoU；
确定所述第五IoU和所述第六IoU中的IoU均不满足所述预设条件，则，为所述第三测试框的轨迹标识设置为目标轨迹标识，所述评估结...
【专利技术属性】
技术研发人员：王姣，刘圣，
申请(专利权)人：北京爱笔科技有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。