Skip to content

Latest commit

 

History

History
47 lines (34 loc) · 2.57 KB

object_track.md

File metadata and controls

47 lines (34 loc) · 2.57 KB
title
MaixCAM MaixPy 物体轨迹追踪和计数(如人流计数)

轨迹追踪简介

前面我们使用 YOLOv5 YOLOv8 甚至是 find_blobs 都可以检测到物体,但是如果画面中同时存在多个物体,当我们需要区分每一个物体,就需要物体追踪功能了。

比如画面中同时有 5 个人在移动,我们需要给每个人编号,知道他们的行动轨迹。

应用:

  • 人流计数,比如通过某个地段的人数量。
  • 工件计数,比如流水线对生产的产品进行计数。
  • 物体移动轨迹记录和识别。

MaixCAM/MaixPy 物体追踪和人流计数效果

效果如下视频,可以跟踪每个人,以及对从上往下跨越黄色区域的人进行计数(左下角):

MaixCAM / MaixPy 使用 物体追踪和人流计数

可以参考直接安装应用 体验。 可以看examples/vision/tracker 下的例程

其中tracker_bytetrack.py 例程是基本的物体跟踪例程,分为几个步骤:

  • 使用 YOLOv5 或者 YOLOv8 检测物体,这样你就可以根据你自己要检测的物体更换模型即可检测不同物体。
  • 使用maix.tracker.ByteTracker 这个算法进行物体追踪,只需要调用一个update函数即可得到结果(画面中的每个轨迹),十分简单。

其中有几个参数根据自己的实际场景进行调整,具体参数以例程代码和 API 参数说明为准:

# configs
conf_threshold = 0.3       # detect threshold
iou_threshold = 0.45       # detect iou threshold
max_lost_buff_time = 120   # the frames for keep lost tracks.
track_thresh = 0.4         # tracking confidence threshold.
high_thresh = 0.6          # threshold to add to new track.
match_thresh = 0.8         # matching threshold for tracking, e.g. one object in two frame iou < match_thresh we think they are the same obj.
max_history_num = 5        # max tack's position history length.
show_detect = False        # show detect
valid_class_id = [0]       # we used classes index in detect model。

tracker_bytetrack_count.py 例程则增加了人流计数例程,这里为了让例程更加简单,只简单地写了一个判断人从上往下走的计数,即当人处在黄色区域以下,同时轨迹在黄色区域内就认为是从上往下跨越了黄色区域。 实际在你的应用场景可以自己编写相关逻辑。