基于深度学习的公交客流检测算法
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

U121

基金项目:

黑龙江省自然科学基金项目(YQ2022E003).


Deep learning-based bus passenger flow detection algorithm
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对公交客流检测因忽略边缘计算而导致的数据处理延迟和准确性问题, 基于“云-边-端协同架构”提出一种实时且轻量级的公交客流检测算法BPF-DETR. 首先, 采用RT-DETR-r18作为基线算法来提高实时处理能力; 然后, 引入轻量级iRMB模块更新ResNet-18作为特征提取主干, 通过倒置残差结构充分学习乘客目标的长距离特征交互以及小目标的局部特征交互, 在提高轻量性和精度的同时增强算法的边缘适用性; 接着, 引入ASF架构中的SSFF模块和TFE模块构建多尺度特征融合模块MSFM, 以进一步提升所提出算法在多尺度和复杂环境下的检测精度; 最后, 为了验证所提出算法的有效性, 采用基于ROI的图像拼接方法, 以提高数据集的代表性和多样性, 构建公交客流监控数据集进行训练验证. 实验结果表明, BPF-DETR的mAP@0.5为96.4 %, 模型大小为32.6 MB, 均优于目前主流的YOLO系列模型, 相较于基线算法, mAP@0.5提升了1.1 %, 模型大小下降了16 %, 能够满足公交客流检测准确率以及边缘部署轻量化要求.

    Abstract:

    To address the issues of data processing latency and accuracy in bus passenger flow detection due to the neglect of edge computing, a real-time and lightweight bus passenger flow detection algorithm based on a “cloud-edge-end collaborative architecture”, bus passenger flow-DETR (BPF-DETR), is proposed. Firstly, RT-DETR-r18 is adopted as the baseline algorithm to enhance real-time processing capabilities. Secondly, a lightweight iRMB module is introduced to update ResNet-18 as the feature extraction backbone. Through the inverted residual structure, the model effectively captures long-range feature interactions of passenger targets and local feature interactions of small object targets, enhancing both lightness and accuracy while improving edge applicability. Thirdly, the SSFF module and TFE module components from the ASF architecture are integrated to construct the multi-scale feature fusion module (MSFM), further enhancing the detection accuracy under multi-scale and complex environments. Finally, to validate the effectiveness of the algorithm, a ROI-based image stitching method is used to enhance the representativeness and diversity of the dataset, constructing a bus passenger monitoring dataset for training and validation. Experimental results show that the mAP@0.5 of the BPF-DETR is 96.4 %, and the model size is 32.6 MB, which are both superior to current mainstream YOLO series models. In comparison with the baseline algorithm, the mAP@0.5 is improved by 1.1 %, and the model size decreases by 16 %, meeting the accuracy and lightweight deployment requirements for bus passenger flow detection.

    参考文献
    相似文献
    引证文献
引用本文

武慧荣,张敬宜,郭春敏.基于深度学习的公交客流检测算法[J].控制与决策,2025,40(6):1827-1837

复制
相关视频

分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-10-24
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2025-04-30
  • 出版日期: 2025-06-20
文章二维码