对象检测算法,我们前期的文章介绍过了很多算法,包含openCV,YOLOv系列检测算法,基于transformer模型的DETR等等模型,都可以进行对象检测。但是对于对象追踪我们很少介绍过,什么是对象追踪?通俗易懂的说法便是在对象检测的基础上,进行对象的实时追踪。
类似上图图片,我们使用对象检测算法,已经检测到了骑摩托车的人,但是每个骑摩托车的人都是独立的个体,若想实现对象追踪,我们就需要给每个人添加一个标签,或者一个ID,这样在对象检测的基础上,添加对象的ID标签,并能够实时进行对象的追踪,确保对象在可检测范围内,能够被算法实时识别,且ID唯一。openCV库下面有8个对象追踪算法可以使用,这里我们不介绍openCV库对象追踪算法,这里我们使用Torchvision实现Deep SORT对象追踪算法。
要实现Deep SORT对象追踪算法,我们需要先安装deep_sort_realtime库,这里安装此库直接使用pip安装即可。
! pip install deep_sort_realtime
Collecting deep_sort_realtime
Downloading deep_sort_realtime-1.3.2-py3-none-any.whl (8.4 MB)
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 8.4/8.4 MB 18.7 MB/s eta 0:00:00
Requ