• 查询稿件
  • 获取最新论文
  • 知晓行业信息

基于改进YOLOv5s的列车车厢客流密度检测方法研究

Detection method of passenger flow density in train carriage based on improved YOLOv5s

  • 摘要: 针对城市轨道交通(简称:城轨)列车车厢客流密度检测过程中人群密集、乘客间相互遮挡的问题,文章提出一种基于改进YOLOv5s模型的列车车厢客流密度检测方法。设计了基于车载闭路电视监控(CCTV,Closed-Circuit Television)系统监控进行实时目标检测的列车车厢客流密度检测模型;为解决人群密集及遮挡问题,对YOLOv5s进行优化,采用了双向特征金字塔网络(BiFPN ,Bi-directional Feature Pyramid Network)结构加强网络特征融合,设计了一种损失函数计算方法,改进了非极大值抑制(NMS,Non-Maximum Suppression)方法,避免候选框误删除的情况。在标准行人检测数据集和自制地铁车厢乘客数据集上进行实验,结果表明,在两类数据集上,改进模型的检测精度均较原模型有所提升。

     

    Abstract: Aiming at the problem of crowded and serious mutual occlusion among passengers in the process of passenger flow density detection of urban rail transit train carriage, this paper proposed a passenger flow density detection method of train carriage based on improved YOLOv5s model, designed a detection model of passenger flow density in the train compartment based on CCTV (Closed Circuit Television) system monitoring for real-time target detection. In order to solve the problem of crowd density and occlusion, the paper optimized YOLOv5s, used BiFPN (Bi directional Feature Pyramid Network) structure to strengthen network feature fusion, designed a loss function calculation method, and improved NMS (Non Maximum Suppression) method to avoid the false deletion of candidate boxes. The paper conducted experiments on the standard pedestrian detection dataset and the self-made subway carriage passenger data set. The results show that the detection accuracy of the improved model is improved compared with the original model on the two types of datasets.

     

/

返回文章
返回