[发明专利]基于YOLOv4模型的跟踪方法

权利要求书

1.一种基于YOLOv4模型的跟踪方法，其特征在于，包括：

步骤1、将原始图片输入到YOLOv4后，YOLOv4输出检测出来的目标图片，将目标图片分别进行编号形成ID号；其中，如果是第一帧，则直接赋予ID并保存图片和对应的坐标信息和宽高；

步骤2、如果不是第一帧，则计算IOU交并比，如果计算值大于阈值，则赋予新的ID，保存信息，保存图片到newfile中；如果计算值小于阈值，则计算相似度，如果计算值达到阈值，则认定上一帧的图片和当前帧的检测图片是同一个目标，此时删除上一帧的图片并保存当前目标的图片到newfile中；然后，遍历当前帧的下一个目标，并运用相同的方法进行计算；如果计算相似度值超过阈值，则定义当前目标为新目标，赋予新的ID，然后单独保存图和对应的ID以及信息；

步骤3、当当前帧的目标检测完了，发现上一帧仍然存在没检查的目标，则确定该目标丢失，则单独保存图、对应的ID以及信息至lossfile中；

步骤4、遍历newfile中新增的图片和lossfile中丢失的图片，将二者进行相似度计算以判断是否是同一个目标；此时，通过ID或者帧数设定一个阈值，比较当前的ID是否大于设定的ID，如果大于直接删除，反之则保留；

步骤5、按照步骤1至步骤4进行不停的循环检测。