一种基于IOU-Tracker的鱼类跟踪方法和装置

摘要

本发明提供了一种基于IOU－Tracker的鱼类跟踪方法和装置，所述方法包括：输入待检测视频，识别出视频中的多个第一鱼类目标，获得第一矩形框、以及每个矩形框在视频帧中的位置信息；根据视频帧的顺序，对相邻两视频帧进行迭代匹配，直到确定为最后一帧时，输出丢失列表和鱼类目标跟踪信息；其中，在每次迭代匹配中，根据第一矩形框的位置信息进行第一IOU值优先匹配，筛选出第二矩形框，继而位置预测匹配，根据匹配结果，更新丢失列表和鱼类目标跟踪信息。本发明实施例采用了高IOU值匹配策略，能解决鱼类之间的粘连问题，实现对鱼类的实时跟踪，在鱼类目标丢失后能够重新进行跟踪，减少跟踪ID转换次数，具有优良的追踪效果。