近一段时间学习了下deepsort,将github上的代码下下来,调通了后,使用pyqt5 设计了一个简单的UI。
主要实现以下几个功能:
(1)侵入检测,统计特定区域的人流
(2)撞线检测,统计向上向下的人流
(3)单目距离检测,计算简易的速度估测
(4)社交距离检测,实时统计相邻两个检测框的距离
detect实现如下:
deepsort显示界面如下:
程序总体框架如下:
遇到的主要问题如下:
(1)显卡性能较弱,检测速度较慢。
(2) 单目检测距离误差较大
调试的过程中,主要是将yolov5的detect 结果,送入deepsort,再次获得检测结果,格式如[380, 380, 420, 420, 'person', 1, 0.94]等。最后将deep sort的结果,进行分类, 和mask进行比对等。可以伪造一些deepsort 结果,送给后续数据处理,达到分段调试的效果,最后再把整个流程串起来。
简要记录下流程。