传动技术 ›› 2024, Vol. 38 ›› Issue (2): 42-46.

• •    

基于人工智能的公交客流统计系统开发

田宇, 侯佩玉, 张明   

  1. 上海申沃客车有限公司技术中心;北华大学;
  • 出版日期:2024-06-30 发布日期:2024-06-30

  • Online:2024-06-30 Published:2024-06-30

摘要: 针对公交客流统计现阶段存在的问题,即客流统计的准确率以及效率、目标易被遮挡、误检、漏检等,基于公交车内的实时监控视频数据,提出一种基于YOLOv5目标检测算法结合DeepSORT多目标跟踪算法的公交客流统计系统。首先选用基于对人体头部特征的检测及跟踪的方法,统计和筛选了不同时间段、不同天气、不同光照强度下的视频数据以确保数据的多样性和代表性,通过AutoLabelImg半自动化标注工具对人体的头部进行标注。通过YOLOv5算法可以得到目标物的检测框和置信度,DeepSORT算法可以得到目标物的ID,解决了上下客重识别的问题。最后通过点过线计数法完成对目标物的计数。经实验测试,所提出的公交客流统计系统在检测精度达到了95%,召回率也达到了94%,mAP_0.5值达到了0.97,大大地提高了客流统计精度。

关键词: 公交客流统计, 目标检测, 目标跟踪, YOLOv5, DeepSORT

中图分类号: